Skip to content


Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Browse files

fix: permutation test score averages across folds

  • Loading branch information...
commit 4eebf9bfb1b1a09746704c66bd065a7d86985a45 1 parent a509d4c
@satra satra authored
Showing with 5 additions and 6 deletions.
  1. +5 −6 sklearn/
11 sklearn/
@@ -805,13 +805,12 @@ def cross_val_score(estimator, X, y=None, score_func=None, cv=None, n_jobs=1,
def _permutation_test_score(estimator, X, y, cv, score_func):
"""Auxilary function for permutation_test_score"""
- y_test = list()
- y_pred = list()
+ avgscore = []
for train, test in cv:
- y_test.append(y[test])
- y_pred.append([train], y[train]).predict(X[test]))
- return score_func(np.ravel(y_test), np.ravel(y_pred))
+ avgscore.append(score_func(y[test],
+ y[train]).predict(X[test])))
+ return np.mean(avgscore)
def _shuffle(y, labels, random_state):
"""Return a shuffled copy of y eventually shuffle among same labels."""
Please sign in to comment.
Something went wrong with that request. Please try again.