It would be great if there was more documentation/code that showed how to use the various functions, and how to train models.