Skip to content

Commit

Permalink
Remove CFIMDB dataset call
Browse files Browse the repository at this point in the history
  • Loading branch information
JonasLuehrs authored Jun 9, 2023
1 parent c99bdfd commit 25dd324
Showing 1 changed file with 0 additions and 21 deletions.
21 changes: 0 additions & 21 deletions classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

0 comments on commit 25dd324

Please sign in to comment.