分析代码

result_analysis.py

@@ -0,0 +1,39 @@
+
+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")[2]
+        if int(true_tag) == 1:
+            true_count += 1
+        if int(pred_tag) == 1:
+            pred_count += 1
+        if true_tag == pred_tag:
+            accuracy_count += 1
+        total_count += 1
+    print("准确率:\t", (true_count + pred_count) / pred_count)
+    print("召回率:\t", (true_count + pred_count) / true_count)
+    print("Accuracy:\t", accuracy_count / total_count)
+
+
+if __name__ == '__main__':
+    path1 = "summary_tag.txt"
+    path2 = "summary_tag_04.txt"
+    with open(path1, encoding="utf-8") as f1:
+        data1 = f1.readlines()
+
+    with open(path2, encoding="utf-8") as f2:
+        data2 = f2.readlines()
+
+    print("默认参数")
+    analysis(data1)
+    print("贝叶斯优化")
+    analysis(data2)