更改 Yellowbrick 可视化颜色

Changing Yellowbrick visiualization colors

在Yellowbrick 可视化库中,分类报告的默认颜色是红色:从浅到深的色调。有没有办法改变这种颜色并使用绿色、蓝色等?

定义可视化工具时,可以设置颜色序列(cmap)。

viz = ClassificationReport(model,  cmap='PuRd')
viz.score(X_test, y_test)

您可以在

找到不同的颜色序列

http://www.scikit-yb.org/en/latest/api/palettes.html#color-sequences