Learning to Imitate Spatial Organization in Multi-robot Systems