1234567891011121314151617181920212223242526272829303132333435363738 |
- def analysis(data):
- """
- 分析结果
- :param data:
- :return:
- """
- pred_count = 0
- true_count = 0
- accuracy_count = 0
- total_count = 0
- for line in data:
- true_tag = line.split("\t")[0]
- pred_tag = line.split("\t")[1]
- if int(true_tag) == 1:
- true_count += 1
- if int(pred_tag) == 1:
- pred_count += 1
- if int(true_tag) == int(pred_tag) == 1:
- accuracy_count += 1
- total_count += 1
- # print("预测为 1 的数量", pred_count)
- # print("实际为 1 的数量", true_count)
- # print("预测为 1,实际也为 1 的数量", accuracy_count)
- # print("total_video", total_count)
- # print("准确率", accuracy_count / pred_count)
- # print("召回率", accuracy_count / true_count)
- print(total_count, pred_count, accuracy_count, true_count, accuracy_count / true_count, accuracy_count / pred_count)
- print("\n")
- if __name__ == '__main__':
- d = [16, 17, 18, 19, 20, 21]
- for item in d:
- path = "summary_tag_03{}_spider.txt".format(item)
- with open(path, encoding="utf-8") as f2:
- data2 = f2.readlines()
- analysis(data2)
|