验证曲线
采用sklearn库,validation_curve的画法直接参考sklearn中文文档即可。点击此处
代码如下
>>> import numpy as np
>>> from sklearn.model_selection import validation_curve
>>> from sklearn.datasets import load_iris
>>> from sklearn.linear_model import Ridge
>>> np.random.seed(0)
>>> iris = load_iris()
>>> X, y = iris.data, iris.target
>>> indices = np.arange(y.shape[0])
>>> np.random.shuffle(indices)
>>> X, y = X[indices], y[indices]
>>> train_scores, valid_scores = validation_curve(Ridge(), X, y, "alpha",
... np.logspace(-7, 3, 3),
... cv=5)
//## "alpha"是param_name,即estimator(Ridge())中待调整的超参数
//## np.logspace(-7, 3, 3)是para_range,即"alpha"的取值范围
>>> train_scores
array([[0.93..., 0.94..., 0.92