def __init__(self,
space_sample,
cache_preprocessed_data=False,
cache_home=None,
**config_kwargs):
self.config_kwargs = config_kwargs
self.cache_preprocessed_data = cache_preprocessed_data
self.cache_home = cache_home
Estimator.__init__(self, space_sample=space_sample)
def __init__(self,
space_sample,
optimizer,
loss,
metrics,
max_model_size=0,
weights_cache=None,
visualization=False):
self.optimizer = optimizer
self.loss = loss
self.metrics = metrics
self.max_model_size = max_model_size
self.weights_cache = weights_cache
self.visualization = visualization
self.model = self._build_model(space_sample)
Estimator.__init__(self, space_sample=space_sample)
def __init__(self,
task,
space_sample,
data_cleaner_params=None,
cache_dir=None,
clear_cache=True):
self.data_pipeline = None
self.clear_cache = clear_cache
self.cache_dir = self._prepare_cache_dir(cache_dir, clear_cache)
# self.task = task
self.data_cleaner_params = data_cleaner_params
self.gbm_model = None
self.data_clearner = None
self.pipeline_signature = None
self.fit_kwargs = None
Estimator.__init__(self, space_sample=space_sample, task=task)
self._build_model(space_sample)