Merge pull request #1 from successar/successar-patch-1

Update Code to take a split
successar · Nov 22, 2019 · 8a3ab2b · 8a3ab2b
2 parents f0cee50 + 4104117
commit 8a3ab2b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Rationale_model/commands/nazneen_predict.sh b/Rationale_model/commands/nazneen_predict.sh
@@ -1,9 +1,9 @@
 export archive=${output_dir:?"Set output_dir"}
-export TEST_DATA_PATH=${dataset_folder:?"set dataset_folder"}/test.jsonl
+export TEST_DATA_PATH=${dataset_folder:?"set dataset_folder"}/${split:?"set a split"}.jsonl
 export batch_size_saliency=${batch_size_saliency-1} 
 
 allennlp predict \
---output-file $archive/lime_test_prediction.jsonl \
+--output-file ${archive}/lime_${split}_prediction.jsonl \
 --predictor rationale_predictor \
 --include-package Rationale_model \
 --silent \
@@ -12,4 +12,4 @@ allennlp predict \
 -o "{model: {saliency_scorer: {type: \"lime\", threshold: ${threshold:?"Set threshold"}, batch_size: ${batch_size_saliency}}}}" \
 --use-dataset-reader \
 --dataset-reader-choice validation \
-$archive/model.tar.gz $TEST_DATA_PATH
+${archive}/model.tar.gz $TEST_DATA_PATH