Update classifier.py

token-tricksters · Feb 19, 2023 · 9227721 · 9227721
1 parent d55ed8a
commit 9227721
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/classifier.py b/classifier.py
@@ -362,8 +362,8 @@ def get_args():
         dev='data/ids-sst-dev.csv',
         test='data/ids-sst-test-student.csv',
         option=args.option,
-        dev_out = 'predictions/ids-sst-dev-out.csv',
-        test_out = 'predictions/ids-sst-test-out.csv'
+        dev_out = 'predictions/sst-dev-out.csv',
+        test_out = 'predictions/sst-test-out.csv'
     )
 
     train(config)
@@ -383,8 +383,8 @@ def get_args():
         dev='data/ids-cfimdb-dev.csv',
         test='data/ids-cfimdb-test-student.csv',
         option=args.option,
-        dev_out = 'predictions/ids-cfimdb-dev-out.csv',
-        test_out = 'predictions/ids-cfimdb-test-out.csv'
+        dev_out = 'predictions/cfimdb-dev-out.csv',
+        test_out = 'predictions/cfimdb-test-out.csv'
     )
 
     train(config)