From 25dd3244385d46052f00e73112c2bacb6070aff7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonas=20L=C3=BChrs?=
 <58192646+JonasLuehrs@users.noreply.github.com>
Date: Fri, 9 Jun 2023 15:28:11 +0200
Subject: [PATCH] Remove CFIMDB dataset call

---
 classifier.py | 21 ---------------------
 1 file changed, 21 deletions(-)

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)