Update predict.py

DeFacto · Apr 26, 2019 · cb5d527 · cb5d527
1 parent ccdb977
commit cb5d527
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/predict.py b/predict.py
@@ -20,7 +20,7 @@
 test_file = jsonlines.open(test_file)
 test_set = []
 
-claim_id = 14001
+claim_id = 1
 
 wiki_entities = os.listdir(wiki_split_docs_dir)
 for i in range(len(wiki_entities)):
@@ -36,7 +36,7 @@
 	test_set.append(lines)
 
 with jsonlines.open(results_file, mode='w') as writer:
-	for example in test_set[14000:]:
+	for example in test_set:
 		relevant_docs,entities = doc_retrieval.getRelevantDocs(example['claim'],wiki_entities,"StanfordNER",nlp)
 		relevant_docs = list(set(relevant_docs))
 		print(example['claim'])