【发布时间】:2014-12-17 16:51:18
【问题描述】:
请参考以下地址的笔记本
这部分代码,
scores = cross_val_score(LogisticRegression(), X, y, scoring='accuracy', cv=10)
print scores
print scores.mean()
在window 7 64位机器中产生如下错误
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-37-4a10affe67c7> in <module>()
1 # evaluate the model using 10-fold cross-validation
----> 2 scores = cross_val_score(LogisticRegression(), X, y, scoring='accuracy', cv=10)
3 print scores
4 print scores.mean()
C:\Python27\lib\site-packages\sklearn\cross_validation.pyc in cross_val_score(estimator, X, y, scoring, cv, n_jobs, verbose, fit_params, score_func, pre_dispatch)
1140 allow_nans=True, allow_nd=True)
1141
-> 1142 cv = _check_cv(cv, X, y, classifier=is_classifier(estimator))
1143 scorer = check_scoring(estimator, score_func=score_func, scoring=scoring)
1144 # We clone the estimator to make sure that all the folds are
C:\Python27\lib\site-packages\sklearn\cross_validation.pyc in _check_cv(cv, X, y, classifier, warn_mask)
1366 if classifier:
1367 if type_of_target(y) in ['binary', 'multiclass']:
-> 1368 cv = StratifiedKFold(y, cv, indices=needs_indices)
1369 else:
1370 cv = KFold(_num_samples(y), cv, indices=needs_indices)
C:\Python27\lib\site-packages\sklearn\cross_validation.pyc in __init__(self, y, n_folds, indices, shuffle, random_state)
428 for test_fold_idx, per_label_splits in enumerate(zip(*per_label_cvs)):
429 for label, (_, test_split) in zip(unique_labels, per_label_splits):
--> 430 label_test_folds = test_folds[y == label]
431 # the test split can be too big because we used
432 # KFold(max(c, self.n_folds), self.n_folds) instead of
IndexError: too many indices for array
我正在使用 scikit.learn 0.15.2,建议 here 可能是 Windows 7、64 位机器的特定问题。
==============更新==============
我发现下面的代码确实有效
from sklearn.cross_validation import KFold
cv = KFold(X.shape[0], 10, shuffle=True, random_state=33)
scores = cross_val_score(LogisticRegression(), X, y, scoring='accuracy', cv=cv)
print scores
==============更新2=============
似乎由于某些软件包更新,我无法再在我的机器上重现此类错误。如果您在 Windows 7 64 位机器上遇到同样的问题,请告诉我。
【问题讨论】:
-
y的形状是什么? -
有效和无效之间的唯一区别是
cv?X.shape[0] == 6366还有吗? -
@eickenberg
cv=10会尝试分层 10 倍简历,KFold不会。 -
如果其他条件相同,明确输入
cv=StratifiedKFold(y, 10)将是我的下一个诊断步骤。 -
这是您所做的唯一更改吗?因为如果可行,那么 cv=number 也应该(见@larsmans 评论)
标签: python scikit-learn cross-validation