Skip to content

Commit 9ed2708

Browse files
Solve #114 (Allowing users to choose the models they want)
1 parent 0b0ab71 commit 9ed2708

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lazypredict/Supervised.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ def __init__(
225225
self.predictions = predictions
226226
self.models = {}
227227
self.random_state = random_state
228-
self.classifiers = CLASSIFIERS
228+
self.classifiers = classifiers
229229

230230
def fit(self, X_train, X_test, y_train, y_test):
231231
"""Fit Classification algorithms to X_train and y_train, predict and score on X_test, y_test.
@@ -526,7 +526,7 @@ def __init__(
526526
self.predictions = predictions
527527
self.models = {}
528528
self.random_state = random_state
529-
self.regressors = REGRESSORS
529+
self.regressors = regressors
530530

531531
def fit(self, X_train, X_test, y_train, y_test):
532532
"""Fit Regression algorithms to X_train and y_train, predict and score on X_test, y_test.

0 commit comments

Comments
 (0)