I mean changes to the Model class itself. The Model class only has get_input_source and get_target_source methods, so it doesn't allow learning the joint distribution between more than these 2 sets of vectors.

I've hopefully made my docstring more clear about this.