借助包 yellowbrick
代码:
from yellowbrick.classifier import ROCAUC
visualizer = ROCAUC(bbc, classes=["Inert", "Slow", "Fast"]) # 这里的bbc 为你自己的模型的名字
visualizer.fit(X_train, y_train) #拟合训练数据
visualizer.score(X_test, y_test) # 测试
visualizer.show()
效果展示: