Skip to content

Commit 0b0ab71

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

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lazypredict/Supervised.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,13 +217,15 @@ def __init__(
217217
custom_metric=None,
218218
predictions=False,
219219
random_state=42,
220+
classifiers = CLASSIFIERS
220221
):
221222
self.verbose = verbose
222223
self.ignore_warnings = ignore_warnings
223224
self.custom_metric = custom_metric
224225
self.predictions = predictions
225226
self.models = {}
226227
self.random_state = random_state
228+
self.classifiers = CLASSIFIERS
227229

228230
def fit(self, X_train, X_test, y_train, y_test):
229231
"""Fit Classification algorithms to X_train and y_train, predict and score on X_test, y_test.

0 commit comments

Comments
 (0)