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)


if __name__ == '__main__':
    # path0 = "summary.txt"
    # path1 = "summary_tag.txt"
    path2 = "summary_tag_0319.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(path2, encoding="utf-8") as f2:
        data2 = f2.readlines()
    # print("无标签")
    # analysis(data0)
    # print("默认参数")
    # analysis(data1)
    # print("贝叶斯优化")
    analysis(data2)