changed the max nbr of training samples from 7.5k to 10k. This will make the training slower, but will improve the results.