finished train and test of the classifier