Pass dimensions to generate_estimator
It's needed in order to create isotropic kernels for the GaussianProcessRegressor
This commit is contained in:
@@ -91,5 +91,5 @@ class HyperOptAuto(IHyperOpt):
|
||||
def trailing_space(self) -> List['Dimension']:
|
||||
return self._get_func('trailing_space')()
|
||||
|
||||
def generate_estimator(self) -> EstimatorType:
|
||||
return self._get_func('generate_estimator')()
|
||||
def generate_estimator(self, dimensions) -> EstimatorType:
|
||||
return self._get_func('generate_estimator')(dimensions)
|
||||
|
||||
Reference in New Issue
Block a user