Cleaning Predictions into one Folder

DeFacto · Feb 24, 2020 · 525e50d · 525e50d
1 parent 7a84b15
commit 525e50d
Show file tree

Hide file tree

Showing 7 changed files with 4 additions and 4 deletions.
diff --git a/predict.py b/predict.py
@@ -11,8 +11,8 @@
 
 nlp = spacy.load('en_core_web_lg')
 
-test_file = "data/shared_task_dev_public_relevant_docs.jsonl"
-results_file = "predictions_dev.jsonl"
+test_file = "data/dev.jsonl"
+results_file = "predictions/predictions_dev_sanity.jsonl"
 
 wiki_dir = 'data/wiki-pages/wiki-pages'
 wiki_split_docs_dir = "data/wiki-pages/wiki-pages-split"

diff --git a/new_predictions.jsonl → predictions/new_predictions.jsonl b/new_predictions.jsonl → predictions/new_predictions.jsonl
diff --git a/predictions.jsonl → predictions/predictions.jsonl b/predictions.jsonl → predictions/predictions.jsonl
diff --git a/predictions_dev.jsonl → predictions/predictions_dev.jsonl b/predictions_dev.jsonl → predictions/predictions_dev.jsonl
diff --git a/predictions_train.jsonl → predictions/predictions_train.jsonl b/predictions_train.jsonl → predictions/predictions_train.jsonl
diff --git a/read_dataset.py b/read_dataset.py
@@ -12,7 +12,7 @@
 train_file = "data/train.jsonl"
 test_file = "data/shared_task_dev_public.jsonl"
 dev_file = "data/dev.jsonl"
-results_file = "predictions.jsonl"
+results_file = "predictions/predictions.jsonl"
 
 wiki_dir = 'data/wiki-pages/wiki-pages'
 wiki_split_docs_dir = "data/wiki-pages/wiki-pages-split"

diff --git a/train_label_classifier.py b/train_label_classifier.py
@@ -188,7 +188,7 @@ def predict_test(predictions_test,entailment_predictions_test,new_predictions_fi
 
 predictions_train = "predictions_train.jsonl"
 predictions_test = "predictions.jsonl"
-new_predictions_file = "new_predictions.jsonl"
+new_predictions_file = "predictions/new_predictions.jsonl"
 
 gold_train = "data/subsample_train_relevant_docs.jsonl"
 entailment_predictions_train = "rte/entailment_predictions_train"