分析代码

result_analysis.py

@@ -16,31 +16,29 @@ def analysis(data):
             true_count += 1
         if int(pred_tag) == 1:
             pred_count += 1
-        if true_tag == pred_tag:
+        if true_tag == pred_tag == 1:
             accuracy_count += 1
         total_count += 1
-    print("视频数量")
-    print("预测为 1", pred_count)
-    print("实际为 1", true_count)
-    print("准确率:\t", (true_count + pred_count) / pred_count)
-    print("召回率:\t", (true_count + pred_count) / true_count)
-    print("Accuracy:\t", accuracy_count / total_count)
+    print("预测为 1 的数量", pred_count)
+    print("实际为 1 的数量", true_count)
+    print("预测为 1，实际也为 1 的数量", accuracy_count)
+    print("total_video", true_count)
 
 
 if __name__ == '__main__':
-    path0 = "summary.txt"
-    path1 = "summary_tag.txt"
-    path2 = "summary_tag_04.txt"
+    # path0 = "summary.txt"
+    # path1 = "summary_tag.txt"
+    path2 = "summary_tag_0321.txt"
 
-    with open(path0, encoding="utf-8") as f0:
-        data0 = f0.readlines()
-    with open(path1, encoding="utf-8") as f1:
-        data1 = f1.readlines()
+    # with open(path0, encoding="utf-8") as f0:
+    #     data0 = f0.readlines()
+    # with open(path1, encoding="utf-8") as f1:
+    #     data1 = f1.readlines()
     with open(path2, encoding="utf-8") as f2:
         data2 = f2.readlines()
-    print("无标签")
-    analysis(data0)
-    print("默认参数")
-    analysis(data1)
-    print("贝叶斯优化")
+    # print("无标签")
+    # analysis(data0)
+    # print("默认参数")
+    # analysis(data1)
+    # print("贝叶斯优化")
     analysis(data2)