Skip to content

Commit

Permalink
Simplify lightgbm_integration.py.
Browse files Browse the repository at this point in the history
  • Loading branch information
g-votte committed Oct 8, 2019
1 parent 6b77bd2 commit 4f02e08
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions examples/pruning/lightgbm_integration.py
Expand Up @@ -18,9 +18,6 @@
import optuna


EPS = 1e-12


# FYI: Objective functions can take additional arguments
# (https://optuna.readthedocs.io/en/stable/faq.html#objective-func-additional-args).
def objective(trial):
Expand All @@ -37,8 +34,8 @@ def objective(trial):
'lambda_l1': trial.suggest_loguniform('lambda_l1', 1e-8, 10.0),
'lambda_l2': trial.suggest_loguniform('lambda_l2', 1e-8, 10.0),
'num_leaves': trial.suggest_int('num_leaves', 2, 256),
'feature_fraction': min(trial.suggest_uniform('feature_fraction', 0.4, 1.0 + EPS), 1.0),
'bagging_fraction': min(trial.suggest_uniform('bagging_fraction', 0.4, 1.0 + EPS), 1.0),
'feature_fraction': trial.suggest_uniform('feature_fraction', 0.4, 1.0),
'bagging_fraction': trial.suggest_uniform('bagging_fraction', 0.4, 1.0),
'bagging_freq': trial.suggest_int('bagging_freq', 1, 7),
'min_child_samples': trial.suggest_int('min_child_samples', 5, 100),
}
Expand Down

0 comments on commit 4f02e08

Please sign in to comment.