问题描述
我正在使用黄砖来绘制 AUCROC。我想从情节中删除标题,使其在没有情节标题的情况下为空。
model = classifier
visualizer = ROCAUC(model,encoder={0: 'class',1: 'class2'}
visualizer.fit(X_train,y_train)
visualizer.score(X_test,y_test)
visualizer.show()
解决方法
yellowbrick 文档 How can I change the title of a Yellowbrick plot?
如果我在 title=" "
中使用单个空格,那么我会得到没有标题的情节。
它不适用于空字符串 title=""
。
最小工作示例
from yellowbrick.classifier import ROCAUC
from sklearn.linear_model import RidgeClassifier
# single space in title
visualizer = ROCAUC(RidgeClassifier(),title=" ")
visualizer.show()