有没有办法用python创建动态混淆矩阵

时间:2019-05-31 15:47:05

标签: python confusion-matrix

我正在尝试使用python计算动态混淆矩阵。这样就可以通过滑块或文本块(也许使用tkinter?)动态更改阈值,并获得更新的混淆矩阵。

目前,我正在使用seaborn热图来显示我的数据分布:

def cmpanda(true,pred,t): 
    y_predtab =[]
    for i in range(len(pred)):
        if pred[i]>t:
            y_predtab.append(1)
        else:
            y_predtab.append(0)
    Data = {'Veri':true[:],'Proba':y_predtab[:]}
    df=pd.DataFrame(Data, columns=['Veri','Proba'])
    confusion_matrix=pd.crosstab(df['Veri'],df['Proba'],rownames=['Actual'], colnames=['Predicted'], margins = False)
    sns.heatmap(confusion_matrix, annot=True)
    cm=ConfusionMatrix(true, y_predtab)
    cm.print_stats()

我得到:confusion matrix

我已经研究过tkinter,但无法设法在其上显示seaboarn热图。只要我可以动态更改阈值,就可以使用更经典的混淆矩阵(2D数组)。

0 个答案:

没有答案