some more tests with retraining
dataset analysis