diff --git a/classifier.py b/classifier.py index 8a0affd..8f43684 100644 --- a/classifier.py +++ b/classifier.py @@ -370,24 +370,3 @@ def get_args(): print('Evaluating on SST...') test(config) - - print('Training Sentiment Classifier on cfimdb...') - config = SimpleNamespace( - filepath='cfimdb-classifier.pt', - lr=args.lr, - use_gpu=args.use_gpu, - epochs=args.epochs, - batch_size=8, - hidden_dropout_prob=args.hidden_dropout_prob, - train='data/ids-cfimdb-train.csv', - dev='data/ids-cfimdb-dev.csv', - test='data/ids-cfimdb-test-student.csv', - option=args.option, - dev_out = 'predictions/'+args.option+'-cfimdb-dev-out.csv', - test_out = 'predictions/'+args.option+'-cfimdb-test-out.csv' - ) - - train(config) - - print('Evaluating on cfimdb...') - test(config)