Advertisement

Python制作混淆矩阵(Python版本)

阅读量:
复制代码
    # ------------------------------------绘制混淆矩阵--------------------------------------
    import pandas as pd
    import matplotlib.pyplot as plt
    import seaborn as sn # 可视化包
    import numpy as np
    Confusion_Matrix = [[0 for i in range(8)] for i in range(8)]  # 整型数组
    # 更新混淆矩阵
    Confusion_Matrix[pred][target_first] = Confusion_Matrix[pred][target_first] + 1
    # 一个DataFrame表示一个表格,类似电子表格的数据结构,包含一个经过排序的列表集
    Confusion_Matrix = np.array(Confusion_Matrix)
    # Confusion_Matrix = Confusion_Matrix.astype(np.float64)
    df_cm = pd.DataFrame(Confusion_Matrix,     # 类似excel,是一种二维表

全部评论 (0)

还没有任何评论哟~