diff --git a/examples/hyperparameter/LightGBM/hyperparameter_158.py b/examples/hyperparameter/LightGBM/hyperparameter_158.py index 7520390a68f..42302390b25 100644 --- a/examples/hyperparameter/LightGBM/hyperparameter_158.py +++ b/examples/hyperparameter/LightGBM/hyperparameter_158.py @@ -13,15 +13,15 @@ def objective(trial): "module_path": "qlib.contrib.model.gbdt", "kwargs": { "loss": "mse", - "colsample_bytree": trial.suggest_uniform("colsample_bytree", 0.5, 1), - "learning_rate": trial.suggest_uniform("learning_rate", 0, 1), - "subsample": trial.suggest_uniform("subsample", 0, 1), - "lambda_l1": trial.suggest_loguniform("lambda_l1", 1e-8, 1e4), - "lambda_l2": trial.suggest_loguniform("lambda_l2", 1e-8, 1e4), + "colsample_bytree": trial.suggest_float("colsample_bytree", 0.5, 1), + "learning_rate": trial.suggest_float("learning_rate", 0, 1), + "subsample": trial.suggest_float("subsample", 0, 1), + "lambda_l1": trial.suggest_float("lambda_l1", 1e-8, 1e4, log=True), + "lambda_l2": trial.suggest_float("lambda_l2", 1e-8, 1e4, log=True), "max_depth": 10, "num_leaves": trial.suggest_int("num_leaves", 1, 1024), - "feature_fraction": trial.suggest_uniform("feature_fraction", 0.4, 1.0), - "bagging_fraction": trial.suggest_uniform("bagging_fraction", 0.4, 1.0), + "feature_fraction": trial.suggest_float("feature_fraction", 0.4, 1.0), + "bagging_fraction": trial.suggest_float("bagging_fraction", 0.4, 1.0), "bagging_freq": trial.suggest_int("bagging_freq", 1, 7), "min_data_in_leaf": trial.suggest_int("min_data_in_leaf", 1, 50), "min_child_samples": trial.suggest_int("min_child_samples", 5, 100), @@ -41,5 +41,5 @@ def objective(trial): dataset = init_instance_by_config(CSI300_DATASET_CONFIG) - study = optuna.Study(study_name="LGBM_158", storage="sqlite:///db.sqlite3") + study = optuna.create_study(study_name="LGBM_158", storage="sqlite:///db.sqlite3") study.optimize(objective, n_jobs=6) diff --git a/examples/hyperparameter/LightGBM/hyperparameter_360.py b/examples/hyperparameter/LightGBM/hyperparameter_360.py index 7ba28c78fe8..495501bceae 100644 --- a/examples/hyperparameter/LightGBM/hyperparameter_360.py +++ b/examples/hyperparameter/LightGBM/hyperparameter_360.py @@ -15,15 +15,15 @@ def objective(trial): "module_path": "qlib.contrib.model.gbdt", "kwargs": { "loss": "mse", - "colsample_bytree": trial.suggest_uniform("colsample_bytree", 0.5, 1), - "learning_rate": trial.suggest_uniform("learning_rate", 0, 1), - "subsample": trial.suggest_uniform("subsample", 0, 1), - "lambda_l1": trial.suggest_loguniform("lambda_l1", 1e-8, 1e4), - "lambda_l2": trial.suggest_loguniform("lambda_l2", 1e-8, 1e4), + "colsample_bytree": trial.suggest_float("colsample_bytree", 0.5, 1), + "learning_rate": trial.suggest_float("learning_rate", 0, 1), + "subsample": trial.suggest_float("subsample", 0, 1), + "lambda_l1": trial.suggest_float("lambda_l1", 1e-8, 1e4, log=True), + "lambda_l2": trial.suggest_float("lambda_l2", 1e-8, 1e4, log=True), "max_depth": 10, "num_leaves": trial.suggest_int("num_leaves", 1, 1024), - "feature_fraction": trial.suggest_uniform("feature_fraction", 0.4, 1.0), - "bagging_fraction": trial.suggest_uniform("bagging_fraction", 0.4, 1.0), + "feature_fraction": trial.suggest_float("feature_fraction", 0.4, 1.0), + "bagging_fraction": trial.suggest_float("bagging_fraction", 0.4, 1.0), "bagging_freq": trial.suggest_int("bagging_freq", 1, 7), "min_data_in_leaf": trial.suggest_int("min_data_in_leaf", 1, 50), "min_child_samples": trial.suggest_int("min_child_samples", 5, 100), @@ -44,5 +44,5 @@ def objective(trial): dataset = init_instance_by_config(DATASET_CONFIG) - study = optuna.Study(study_name="LGBM_360", storage="sqlite:///db.sqlite3") + study = optuna.create_study(study_name="LGBM_360", storage="sqlite:///db.sqlite3") study.optimize(objective, n_jobs=6)