Skip to content

Commit d75e0c2

Browse files
mohatahermohataher
mohataher
authored and
mohataher
committed
fixed bug at ClimbsRocks#417
1 parent 4b7e0e7 commit d75e0c2

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

Diff for: auto_ml/predictor.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,11 @@ def _pickle_method(m):
5959
copyreg.pickle(types.MethodType, _pickle_method)
6060

6161

62+
class ExtendedEvolutionaryAlgorithmSearchCV(EvolutionaryAlgorithmSearchCV):
63+
def _run_search(cls):
64+
"""required method on base class but not implemented; it causes a TypeError"""
65+
pass
66+
6267
class Predictor(object):
6368

6469

@@ -1131,7 +1136,7 @@ def fit_grid_search(self, X_df, y, gs_params, feature_learning=False, refit=Fals
11311136
fit_evolutionary_search = True
11321137
# For some reason, EASCV doesn't play nicely with CatBoost. It blows up the memory hugely, and takes forever to train
11331138
if fit_evolutionary_search == True:
1134-
gs = EvolutionaryAlgorithmSearchCV(
1139+
gs = ExtendedEvolutionaryAlgorithmSearchCV(
11351140
# Fit on the pipeline.
11361141
ppl,
11371142
# Two splits of cross-validation, by default

0 commit comments

Comments
 (0)