yellowbrick中ClassificationReport多分类图标一行数据异常全为0

发布时间 2023-12-02 23:05:06作者: 扬中源

from yellowbrick.classifier import ClassificationReport
visualizer = ClassificationReport(model, support=True)
visualizer.fit(X_train_std, Y_train)# 拟合visualizer
visualizer.score(X_test_std, Y_test)# 评价模型在训练集上效果
visualizer.show()

 

yellowbrick中ClassificationReport多分类图标一行数据异常全为0,可能是分类标签的问题。

y标签如果是3,1,0就会出现分类为3的整行数据全为0,把标签3全部替换为2即正常显示

y=y.str.replace('3', '2')