|
@@ -217,7 +217,7 @@ if __name__ == "__main__":
|
|
|
# # L.evaluate_model()
|
|
|
# L.feature_importance()
|
|
|
L = LightGBM("train", "whole")
|
|
|
- study = optuna.create_study(direction=clear'maximize')
|
|
|
+ study = optuna.create_study(direction='maximize')
|
|
|
study.optimize(L.bays_params, n_trials=100)
|
|
|
print('Number of finished trials:', len(study.trials))
|
|
|
print('Best trial:', study.best_trial.params)
|