【发布时间】:2018-10-04 21:08:14
【问题描述】:
我有以下代码,我想将k-fold cross validation 用于线性回归模型:
kf = KFold(n_splits=100)
predi = cross_val_predict(model, train[columns], train[target], cv = kf)
predi = pandas.Series(predi)
model.fit(data[columns], data[target])
pred_test = model.predict(test[columns])
print(mean_squared_error(pred_test, test[target]))
但是,我不确定代码是否符合我的要求。具体来说,我不确定model.fit 部分。它甚至使用交叉验证吗?
我不确定这样计算会产生比没有交叉验证更糟糕的结果的原因。
【问题讨论】:
标签: python machine-learning scikit-learn