1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
| import xlrd import xlwt
@main.route('/get_testset_score', methods=['GET', 'POST']) def get_api(): data = xlrd.open_workbook('/data/python_server/code/wanghaodi/lalabot/data/test/机器人测试集.xls') write_content = xlwt.Workbook(encoding='utf8') table = data.sheet_by_index(0) test_list = [] for x, y in zip(table.col_values(0), table.col_values(1)): print(x, y) u = User("lalabot", "123456") query = x type = y session = Session(u, query) session.select_skill_and_process()
skill_name = session.skill.__class__.__name__ if type.find("/"): skill_list_label = type.split("/") score_list = [] for skill_label in skill_list_label: if session.skill.__class__.__name__ == skill_label: score = "1" else: score = "0" score_list.append(score) if "1" in score_list: test = session.get_last_interaction().user_query + "|" + skill_name + "|" + "1" else: test = session.get_last_interaction().user_query + "|" + skill_name + "|" + "0" test_list.append(test) else: if skill_name == type: test = session.get_last_interaction().user_query + "|" + skill_name + "|" + "1" else: test = session.get_last_interaction().user_query + "|" + skill_name + "|" + "0" test_list.append(test)
print(test_list)
write_Excel = write_content.add_sheet('打分结果', cell_overwrite_ok=True)
font = xlwt.Font() font.name = '微软雅黑'
font.height = 200
style = xlwt.XFStyle() style.font = font
write_Excel.write(0, 0, "query", style) write_Excel.write(0, 1, "type", style) write_Excel.write(0, 2, "score", style) s = 1 for i in test_list: write_Excel.write(s, 0, i.split("|")[0], style) write_Excel.write(s, 1, i.split("|")[1], style) write_Excel.write(s, 2, i.split("|")[2], style) s = s + 1 write_content.save('/data/python_server/code/wanghaodi/lalabot/data/test/机器人QA评分v1.xls') return "test"
|