@@ -163,7 +163,7 @@ class LazyClassifier:
163163 prediction : bool, optional (default=False)
164164 When set to True, the predictions of all the models models are returned as dataframe.
165165 classifiers : list, optional (default="all")
166- When function is provided, trains the chosen classifier(s).
166+ When function is provided, trains the chosen classifier(s).
167167
168168 Examples
169169 --------
@@ -285,17 +285,18 @@ def fit(self, X_train, X_test, y_train, y_test):
285285 )
286286
287287 if self .classifiers == "all" :
288- self .classifiers = CLASSIFIERS
288+ self .classifiers = CLASSIFIERS
289289 else :
290- try :
291- temp_list = []
292- for classifier in self .classifiers :
293- full_name = (classifier .__class__ .__name__ , classifier )
294- temp_list .append (full_name )
295- self .classifiers = temp_list
296- except Exception as exception :
297- print (exception )
298- print ("Invalid Classifier(s)" )
290+ try :
291+ temp_list = []
292+ for classifier in self .classifiers :
293+ full_name = (classifier .__class__ .__name__ , classifier )
294+ temp_list .append (full_name )
295+ self .classifiers = temp_list
296+ except Exception as exception :
297+ print (exception )
298+ print ("Invalid Classifier(s)" )
299+
299300 for name , model in tqdm (self .classifiers ):
300301 start = time .time ()
301302 try :
@@ -446,7 +447,7 @@ class LazyRegressor:
446447 prediction : bool, optional (default=False)
447448 When set to True, the predictions of all the models models are returned as dataframe.
448449 regressors : list, optional (default="all")
449- When function is provided, trains the chosen regressor(s).
450+ When function is provided, trains the chosen regressor(s).
450451
451452 Examples
452453 --------
@@ -586,17 +587,17 @@ def fit(self, X_train, X_test, y_train, y_test):
586587 )
587588
588589 if self .regressors == "all" :
589- self .regressors = REGRESSORS
590+ self .regressors = REGRESSORS
590591 else :
591- try :
592- temp_list = []
593- for regressor in self .regressors :
594- full_name = (regressor .__class__ .__name__ , regressor )
595- temp_list .append (full_name )
596- self .regressors = temp_list
597- except Exception as exception :
598- print (exception )
599- print ("Invalid Regressor(s)" )
592+ try :
593+ temp_list = []
594+ for regressor in self .regressors :
595+ full_name = (regressor .__class__ .__name__ , regressor )
596+ temp_list .append (full_name )
597+ self .regressors = temp_list
598+ except Exception as exception :
599+ print (exception )
600+ print ("Invalid Regressor(s)" )
600601
601602 for name , model in tqdm (self .regressors ):
602603 start = time .time ()
0 commit comments