[docs]
1class GridSearchMixin(object):
[docs]
2 def gridsearch(self, Xname, Yname=None, parameters=None, pure_python=False, **kwargs):
3 """ run gridsearch on model
4
5 Args:
6 Xname (str|obj): the name of the X dataset in om.datasets, or
7 the data object
8 Yname (str|obj): the name of the Y dataset in om.datasets, or
9 the data object
10 parameters (dict): input to GridSearchCV(..., param_grid=parameters)
11
12 See Also:
13 * sklearn.model_selection.GridSearchCV
14 """
15 gs_task = self.task('omegaml.tasks.omega_gridsearch')
16 Xname = self._ensure_data_is_stored(Xname, prefix='_fitX')
17 if Yname is not None:
18 Yname = self._ensure_data_is_stored(Yname, prefix='_fitY')
19 return gs_task.delay(self.modelname, Xname, Yname=Yname, parameters=parameters, **kwargs)