Self-Organizing Generative Models For Diverse Imitation Learning