Update train_classifier.ipynb