Skip to content

Commit

Permalink
bug fix when using refit=False (#385)
Browse files Browse the repository at this point in the history
bug fix when using refit=False
  • Loading branch information
selay01 authored and rasbt committed May 20, 2018
1 parent 8a5db64 commit a02c344
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mlxtend/regressor/stacking_cv_regression.py
Expand Up @@ -140,8 +140,8 @@ def fit(self, X, y, groups=None):
self.regr_ = [clone(clf) for clf in self.regressors]
self.meta_regr_ = clone(self.meta_regressor)
else:
self.clfs_ = self.regressors
self.meta_clf_ = self.meta_regressor
self.regr_ = self.regressors
self.meta_regr_ = self.meta_regressor

kfold = check_cv(self.cv, y)
if isinstance(self.cv, int):
Expand Down

0 comments on commit a02c344

Please sign in to comment.