From d75e0c2bbc1c6fbe377032647c794579c1c960db Mon Sep 17 00:00:00 2001 From: mohataher <7adidSingGithub> Date: Thu, 1 Nov 2018 10:13:06 +0000 Subject: [PATCH] fixed bug at #417 --- auto_ml/predictor.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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