python - Yellowbrick learningCurve:chage legend
问题描述
我想使用黄砖学习曲线绘制训练集和验证集的学习曲线。在绘制学习曲线时,我没有使用交叉验证,而是使用了一个保留验证集。无论如何,图例固定为“交叉验证分数”。
有没有办法用另一个字符串替换它?
解决方案
Viz
在 ax 属性中返回轴对象,您可以使用它来设置标签,如下所示:
viz.ax.get_lines()[1].set_label('My custom label')
示例:
import numpy as np
from yellowbrick.datasets import load_energy
from yellowbrick.model_selection import ValidationCurve
from sklearn.tree import DecisionTreeRegressor
# Load a regression dataset
X, y = load_energy()
viz = ValidationCurve(
DecisionTreeRegressor(), param_name="max_depth",
param_range=np.arange(1, 11), cv=10, scoring="r2"
)
# Fit and show the visualizer
viz.fit(X, y)
viz.ax.get_lines()[1].set_label('My custom label')
viz.show()
推荐阅读
- node.js - node.js 的猫鼬外键问题
- matlab - 在matlab中检查没有for循环的数组中的成员资格
- int - 为什么我的 sprintf 与 USB 通信不能很好地工作?
- c - 布尔值,如果不是 0 或 1,则运行时错误
- javascript - Angular 8迭代非空数组中的问题
- python - 如何在 python 中调用 Post API?
- android - 在 Pepper QISDK 上创建/写入文件
- spring-cloud-dataflow - SCDF Spring Cloud Data Fflow 问题与 task-launcher-dataflow 检测到 Data Flow 服务器版本不兼容
- laravel - 如何将 Laravel 应用程序连接到 Lumen api 服务?
- sql - 在“当匹配时”上使用插入和更新 // 合并