make epoch an attribute of model, adjust train_classifier
make epoch an attribute of model, adjust train_classifier