wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

avoids parsing exceptions (#88)

Browse Source 
* avoids parsing exceptions

* remove unnecessary print
This commit is contained in:
rosequ
2017-12-02 10:23:40 -05:00
committed by GitHub
parent ad69d3abc2
commit aee1541722
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
anserini_dependency/RetrieveSentences.py
+2 -3
View File
@@ -60,7 +60,7 @@ class RetrieveSentences:
The number of passages to be returned
"""
scorer = self.rs.getRankedPassagesList(query, index, int(hits), int(k))
scorer = self.rs.getRankedPassagesList(query.strip(), index, int(hits), int(k))
candidate_passages_scores = []
for i in range(0, scorer.size()):
candidate_passages_scores.append(scorer.get(i))
@@ -78,7 +78,7 @@ if __name__ == "__main__":
parser.add_argument("-index", help="Lucene index", required=True)
parser.add_argument("-embeddings", help="Path of the word2vec index", default="")
parser.add_argument("-topics", help="topics file", default="")
parser.add_argument("-query", help="a single query", default="")
parser.add_argument("-query", help="a single query", default="who is newton")
parser.add_argument("-hits", help="max number of hits to return", default=100)
parser.add_argument("-scorer", help="passage scores", default="Idf")
parser.add_argument("-k", help="top-k passages to be retrieved", default=1)
@@ -87,4 +87,3 @@ if __name__ == "__main__":
rs = RetrieveSentences(args_raw)
sc = rs.getRankedPassages(args_raw.query, args_raw.index, args_raw.hits, args_raw.k)