Yellowbrick 学习曲线：改变图例

【问题标题】：Yellowbrick learningCurve: chage legendYellowbrick 学习曲线：改变图例
【发布时间】：2020-11-06 01:50:30
【问题描述】：

我想使用黄砖学习曲线绘制训练集和验证集的学习曲线。在绘制学习曲线时，我没有使用交叉验证，而是使用了一个保留验证集。无论如何，图例固定为“交叉验证分数”。

有没有办法用另一个字符串替换它？

【问题讨论】：

【解决方案1】：

Viz 在 ax 属性中返回坐标轴对象，您可以使用它来设置标签：

viz.ax.get_lines()[1].set_label('My custom label')

Example:

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()

【讨论】：

非常感谢，这正是我需要知道的！