Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, objective='regression', metric='mae', boosting='gbdt', params=params, space=space,
n_est=500, n_stop=10, sample_size=SAMPLE_SIZE, feature_selection=True, n_fs=10,
hyperparam_opt=True, n_hpopt=100, n_random_col=10, random_state=RANDOM_SEED, shuffle=True):
self.metric, minimize = self._get_metric_alias_minimize(metric)
params = self.params
params.update({'objective': objective,
'metric': self.metric,
'boosting': boosting})
super(AutoLGB, self).__init__(params=params, space=space, n_est=n_est, n_stop=n_stop,
sample_size=sample_size, feature_selection=feature_selection,
n_fs=n_fs, hyperparam_opt=hyperparam_opt, n_hpopt=n_hpopt,
minimize=minimize, n_random_col=n_random_col, random_state=random_state,
shuffle=shuffle)