From 251681501020c6b48613d0c6e0c736d6151d0db6 Mon Sep 17 00:00:00 2001 From: Toshihiko Yanase Date: Thu, 8 Sep 2022 10:10:44 +0900 Subject: [PATCH] Use suggest_float instead of suggest_int --- lm_optimizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lm_optimizer.py b/lm_optimizer.py index 25d8a05eb2..f114b024bc 100644 --- a/lm_optimizer.py +++ b/lm_optimizer.py @@ -24,8 +24,8 @@ def character_based(): return is_character_based def objective(trial): - FLAGS.lm_alpha = trial.suggest_uniform('lm_alpha', 0, FLAGS.lm_alpha_max) - FLAGS.lm_beta = trial.suggest_uniform('lm_beta', 0, FLAGS.lm_beta_max) + FLAGS.lm_alpha = trial.suggest_float('lm_alpha', 0, FLAGS.lm_alpha_max) + FLAGS.lm_beta = trial.suggest_float('lm_beta', 0, FLAGS.lm_beta_max) is_character_based = trial.study.user_attrs['is_character_based']