alg/result_analysis.py

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("\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)