10折交叉验证python单变量回归

时间:2016-02-18 05:13:48

标签: python machine-learning linear-regression cross-validation

{I}将数据分成10倍时,numpy vstack如何在numpy中工作。

X_set = np.split(X, 10)
Y_set = np.split(Y, 10)
for i in range(len(X_set)):
   X_test= ?
   Y_test= ?

1 个答案:

答案 0 :(得分:2)

你可以做你在Numpy开始做的事情,但我觉得这种东西太低了。我建议您安装sklearn。然后,您可以执行以下操作

from sklearn import cross_validation

for tr, te in cross_validation.KFold(len(Y_set), 10):
    x_train, y_train = X_set[tr], Y_set[tr]
    x_test, y_test = X_set[te], Y_set[te]