Improve get training data

[cfuselli]

The way the training data is selected now can lead to unwillingly use the same data used for training, a better system would be nice also to allow training and testing the model with different data.