如何移除yellowbrick库ROCAUC可视化生成图表的标题
解决方案
你可以通过以下两种方式实现去掉ROCAUC曲线的默认标题:
方法1(推荐):初始化时直接指定空标题
yellowbrick的所有可视化组件都支持在实例化时传入title参数自定义标题内容,直接传空字符串即可完全隐藏标题:
model = classifier # 补充原代码遗漏的右括号,新增title参数设为空 visualizer = ROCAUC(model, encoder={0: 'class' , 1: 'class2'}, title="") visualizer.fit(X_train, y_train) visualizer.score(X_test, y_test) visualizer.show()
方法2:手动操作底层轴对象清空标题
如果已经完成了拟合、评分步骤,也可以直接修改可视化器绑定的matplotlib轴对象移除标题:
model = classifier visualizer = ROCAUC(model, encoder={0: 'class' , 1: 'class2'}) visualizer.fit(X_train, y_train) visualizer.score(X_test, y_test) # 手动清空标题 visualizer.ax.set_title("") visualizer.show()
内容的提问来源于stack exchange,提问作者user
相关产品推荐
相关产品推荐

