改变Yellowbrick visiualization颜色

Gur*_*cor 2 python-3.x yellowbrick

在Yellowbrick visiualization库中,分类报告的默认颜色为红色:从浅色到深色.有没有办法改变这种颜色,并使用绿色,蓝色等?

lar*_*ray 5

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

viz = ClassificationReport(model,  cmap='PuRd')
viz.score(X_test, y_test)
Run Code Online (Sandbox Code Playgroud)

PurpletoRedReport

你可以找到不同的颜色序列

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