Must load all data before training the classifier