wrong_ids[model_name].append(
(true_sourse, found_sourse))
output_answer.append(
dict(true_sourse=true_sourse,
sourse=found_sourse,
model=model_name,
title=answer[model_name][0]['title'],
similarity=answer[model_name][0]['similarity']))
file.write(json.dumps(output_answer, indent=2, ensure_ascii=False))
file.write(json.dumps(wrong_ids, indent=2, ensure_ascii=False))
if __name__ == "__main__":
log_of_result = []
if len(DBController.get_all_documents()) == 0:
print('Seeding database...')
DatabaseSeeder.seed()
lsi = Storage.load_model('out/lsi', 'phyge', 'lsi')
lda = Storage.load_model('out/lda', 'phyge', 'lda')
d2v = Storage.load_model('out/d2v', 'phyge', 'd2v')
fast_text = Storage.load_model('out/fast_text', 'phyge', 'ft')
search_engine = SearchEngine(models=[fast_text, d2v, lda, lsi])
test_path = os.path.join(PhyVariables.testsDir,
'test_' + str(PhyVariables.queriesId))
run_search(os.path.join(test_path, PhyVariables.queriesFileName),
os.path.join(test_path, PhyVariables.answersFileName), 1)
# run_search('Resources/pdf_articles.json','Resources/answers.json',1)
