From 8eb07315f415e624dd27681592904316c3d7205c Mon Sep 17 00:00:00 2001 From: hanshanley Date: Tue, 21 Feb 2023 15:42:44 -0800 Subject: [PATCH] Update classifier.py --- classifier.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/classifier.py b/classifier.py index 7b758e9..8a0affd 100644 --- 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/sst-dev-out.csv', - test_out = 'predictions/sst-test-out.csv' + dev_out = 'predictions/'+args.option+'-sst-dev-out.csv', + test_out = 'predictions/'+args.option+'-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/cfimdb-dev-out.csv', - test_out = 'predictions/cfimdb-test-out.csv' + dev_out = 'predictions/'+args.option+'-cfimdb-dev-out.csv', + test_out = 'predictions/'+args.option+'-cfimdb-test-out.csv' ) train(config)