if len(docids) > 10:
docids = docids[:10] #截取前10
self.results.append((qid, docids))
def output(self):
with open(self.outfile, 'w') as out:
for (qid, docids) in self.results:
for docid in docids:
print(qid, docid, file=out)
#==============================================================================
# MAIN
if __name__ == '__main__':
starttime = timer()
config = CommandLine()
if config.exit:
sys.exit(0)
index = IndexLoader(config.indexFile).getIndex()
retrieve = Retrieve(index, config.termWeighting)
queries = Queries(config.queriesFile)
allResults = ResultStore(config.outfile)
for qid in queries.qids():
query = queries.getQuery(qid)
results = retrieve.forQuery(query)
allResults.store(qid, results)
allResults.output()
print('TIME: %.2fs' % (timer() - starttime))
docids = docids[:10]
self.results.append((qid, docids))
def output(self):
with open(self.outfile, 'w') as out:
for (qid, docids) in self.results:
for docid in docids:
print(qid, docid, file=out)
#==============================================================================
# MAIN
if __name__ == '__main__':
config = CommandLine()
if config.exit:
sys.exit(0)
index = IndexLoader(config.indexFile).getIndex()
retrieve = Retrieve(index,
config.termWeighting) #interaction with my_retriever
queries = Queries(config.queriesFile)
allResults = ResultStore(config.outfile)
for qid in queries.qids():
query = queries.getQuery(qid)
results = retrieve.forQuery(query) #interaction with my_retriever
allResults.store(qid, results)
allResults.output()
print(qid, docid, file=out)
#==============================================================================
# MAIN
if __name__ == '__main__':
config = CommandLine()
if config.exit:
sys.exit(0)
#print(config.indexFile)
index = IndexLoader(config.indexFile).getIndex()
#config.termWeighting is the query type
#print(index)
retrieve = Retrieve(index, config.termWeighting)
#print(retrieve)
#print(config.queriesFile)
queries = Queries(config.queriesFile)
#print(config.outfile)
allResults = ResultStore(config.outfile)
#print(queries.qids())
#print(retrieve.forQuery(queries.getQuery(1)))
t = MyTimer()
t.start('retrieval')
for qid in queries.qids():
#print(qid)
query = queries.getQuery(qid)
#print(query)
def output(self, outfile):
with open(outfile, 'w') as out:
for (qid, docids) in self.results:
for docid in docids:
print(qid, docid, file=out)
#==============================================================================
# MAIN
if __name__ == '__main__':
start = datetime.datetime.now()
config = CommandLine()
if config.exit:
sys.exit(0)
queries = config.queries
retrieve = Retrieve(config.index, config.term_weighting)
all_results = Result_Store()
for (qid, query) in queries:
results = retrieve.for_query(query)
all_results.store(qid, results)
end = datetime.datetime.now()
time = (end - start).total_seconds()
print(time)
print("Finished!")
all_results.output(config.outfile)
