result_analysis.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import os
  2. def analysis(data):
  3. """
  4. 分析结果
  5. :param data:
  6. :return:
  7. """
  8. pred_count = 0
  9. true_count = 0
  10. accuracy_count = 0
  11. total_count = 0
  12. for line in data:
  13. true_tag = line.split("\t")[0]
  14. pred_tag = line.split("\t")[1]
  15. if int(true_tag) == 1:
  16. true_count += 1
  17. if int(pred_tag) == 1:
  18. pred_count += 1
  19. if int(true_tag) == int(pred_tag) == 1:
  20. accuracy_count += 1
  21. total_count += 1
  22. # print("预测为 1 的数量", pred_count)
  23. # print("实际为 1 的数量", true_count)
  24. # print("预测为 1,实际也为 1 的数量", accuracy_count)
  25. # print("total_video", total_count)
  26. # print("准确率", accuracy_count / pred_count)
  27. # print("召回率", accuracy_count / true_count)
  28. print(total_count, pred_count, accuracy_count, true_count, accuracy_count / true_count, accuracy_count / pred_count)
  29. print("\n")
  30. if __name__ == '__main__':
  31. path = "result"
  32. files = os.listdir(path)
  33. for file in files:
  34. if file.endswith(".txt"):
  35. fp = os.path.join(path, file)
  36. with open(path, encoding="utf-8") as f2:
  37. data2 = f2.readlines()
  38. analysis(data2)