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 true_tag == pred_tag:
            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)


if __name__ == '__main__':
    path0 = "summary.txt"
    path1 = "summary_tag.txt"
    path2 = "summary_tag_04.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)