Trained the models!

[NikitaShymberg]

Here are some trained models to get the analysis going. At a glance, they're not performing super well, but this should be good enough for now.

Also, I wrote docstrings for all the functions.