【发布时间】:2021-10-18 05:10:56
【问题描述】:
我正在尝试使用cross_val_score 来评估我的回归模型(使用PolymonialFeatures(degree = 2))。正如我在不同的博客文章中所指出的那样,我应该将 cross_val_score 与原始 X、y 值一起使用,而不是 X_train 和 y_train。
r_squareds = cross_val_score(pipe, X, y, cv=10)
r_squareds
>>> array([ 0.74285583, 0.78710331, -1.67690578, 0.68890253, 0.63120873,
0.74753825, 0.13937611, 0.18794756, -0.12916661, 0.29576638])
这表明我的模型表现不佳,平均 r2 仅为 0.241。这应该是正确的解释吗?
但是,我遇到了一个处理相同数据的 Kaggle 代码,这个人在 X_train 和 y_train 上执行了 cross_val_score。我试了一下,平均 r2 更好。
r_squareds = cross_val_score(pipe, X_train, y_train, cv=10)
r_squareds.mean()
>>> 0.673
这应该是个问题吗?
这是我的模型的代码:
X = df[['CHAS', 'RM', 'LSTAT']]
y = df['MEDV']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=0)
pipe = Pipeline(
steps=[('poly_feature', PolynomialFeatures(degree=2)),
('model', LinearRegression())]
)
## fit the model
pipe.fit(X_train, y_train)
【问题讨论】:
标签: python scikit-learn linear-regression