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)