yellowbrick中ClassificationReport多分類圖標一行數據異常全爲0

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章