Training the classifier