diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py index 9b11ddc..cb0d418 100644 --- a/auto_ml/predictor.py +++ b/auto_ml/predictor.py @@ -59,6 +59,11 @@ def _pickle_method(m): copyreg.pickle(types.MethodType, _pickle_method) +class ExtendedEvolutionaryAlgorithmSearchCV(EvolutionaryAlgorithmSearchCV): + def _run_search(cls): + """required method on base class but not implemented; it causes a TypeError""" + pass + class Predictor(object): @@ -1131,7 +1136,7 @@ def fit_grid_search(self, X_df, y, gs_params, feature_learning=False, refit=Fals fit_evolutionary_search = True # For some reason, EASCV doesn't play nicely with CatBoost. It blows up the memory hugely, and takes forever to train if fit_evolutionary_search == True: - gs = EvolutionaryAlgorithmSearchCV( + gs = ExtendedEvolutionaryAlgorithmSearchCV( # Fit on the pipeline. ppl, # Two splits of cross-validation, by default