Skip to content

Commit 26d3323

Browse files
committed
Remove redundant p variable
1 parent e35c90c commit 26d3323

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

sklearn/cross_validation.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1043,20 +1043,18 @@ def cross_val_predict(estimator, X, y=None, cv=None, n_jobs=1,
10431043
fit_params)
10441044
for train, test in cv)
10451045

1046-
p = [p for p, _ in preds_blocks]
1046+
preds = [p for p, _ in preds_blocks]
10471047
locs = np.concatenate([loc for _, loc in preds_blocks])
1048-
10491048
if not _check_is_partition(locs, _num_samples(X)):
10501049
raise ValueError('cross_val_predict only works for partitions')
10511050
inv_locs = np.empty(len(locs), dtype=int)
10521051
inv_locs[locs] = np.arange(len(locs))
10531052

10541053
# Check for sparse predictions
1055-
if sp.issparse(p[0]) :
1056-
p = sp.vstack(p, format = 'csr')
1054+
if sp.issparse(preds[0]):
1055+
preds = sp.vstack(preds, format=preds[0].format)
10571056
else :
1058-
p = np.concatenate(p)
1059-
preds = p.copy()
1057+
preds = np.concatenate(preds)
10601058
return preds[inv_locs]
10611059

10621060
def _fit_and_predict(estimator, X, y, train, test, verbose, fit_params):

0 commit comments

Comments
 (0)