Added the train function in the OctreeGPMap class