I've reproduced the results and verified that they are fine. However I'd like to run the trained models on some new data, and I can't seem to make that work. Maybe you could point me in the right direction? Is it at all possible without retraining everything, because to me it seems you didn't store any trained models, only their predictions.