diff --git a/igel/igel.py b/igel/igel.py index bb7266d..bb94fff 100644 --- a/igel/igel.py +++ b/igel/igel.py @@ -346,6 +346,7 @@ def fit(self, **kwargs): **cv_params) hyperparams_props = self.model_props.get('hyperparameter_search', None) if hyperparams_props: + # perform hyperparameter search method = hyperparams_props.get('method', None) grid_params = hyperparams_props.get('parameter_grid', None)