eval.py 文件源码

python
阅读 38 收藏 0 点赞 0 评论 0

项目:image-classifier 作者: gustavkkk 项目源码 文件源码
def plot_conf_matrix(y_actual,y_predict,labels):
    cm = confusion_matrix(y_actual,y_predict,labels)
    fig = plt.figure()
    ax = fig.add_subplot(111)
    cax = ax.matshow(cm)
    pl.title('confusion matrix')
    fig.colorbar(cax)
    ax.set_xticklabels([''] + labels)
    ax.set_yticklabels([''] + labels)
    pl.xlabel('Predicted')
    pl.ylabel('True')
    pl.show()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号