Skip to content

Commit

Permalink
fixed bug at #417
Browse files Browse the repository at this point in the history
  • Loading branch information
mohataher authored and mohataher committed Nov 1, 2018
1 parent 4b7e0e7 commit d75e0c2
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion auto_ml/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):


Expand Down Expand Up @@ -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
Expand Down

0 comments on commit d75e0c2

Please sign in to comment.