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')