Yellowbrick 学习曲线:改变图例
Yellowbrick learningCurve: chage legend
我想使用 yellowbrick 学习曲线绘制训练集和验证集的学习曲线。
绘制学习曲线时我没有使用交叉验证,而是使用了验证集。无论如何,图例固定为“交叉验证分数”。
有没有办法用另一个字符串替换它?
Viz
returns ax中的axes对象属性,你可以用它来设置标签:
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()
我想使用 yellowbrick 学习曲线绘制训练集和验证集的学习曲线。 绘制学习曲线时我没有使用交叉验证,而是使用了验证集。无论如何,图例固定为“交叉验证分数”。
有没有办法用另一个字符串替换它?
Viz
returns ax中的axes对象属性,你可以用它来设置标签:
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()