wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update visual regression tests for issue 410

Browse Source
This commit is contained in:
c-bata
2023-05-25 18:47:38 +09:00
parent 129e316f3d
commit 2360f2c3a0
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hack/visual_regression_test.py
+24
View File
@@ -168,6 +168,30 @@ def create_dummy_storage() -> optuna.storages.InMemoryStorage:
study.optimize(objective_single_inf_report, n_trials=50)
# Issue 410
study = optuna.create_study(study_name="Issue 410", storage=storage, sampler=sampler)
def objective_issue_410(trial: optuna.Trial) -> float:
resample_rate = trial.suggest_categorical("resample_rate", ["50ms"])
channels = trial.suggest_categorical("channels", ["all"])
window_size = trial.suggest_categorical("window_size", [256])
if trial.number > 15:
raise Exception("Unexpected error")
cbow = trial.suggest_categorical("cbow", [True])
model_type = trial.suggest_categorical("model", ["m1"])
height = None
trial.set_user_attr("epochs", 0)
trial.set_user_attr("deterministic", True)
if trial.number > 10:
raise Exception("unexpeccted error")
trial.set_user_attr("folder", "/path/to/folder")
trial.set_user_attr("resample_type", "foo")
trial.set_user_attr("run_id", "0001")
return 1.0
study.optimize(objective_issue_410, n_trials=20, catch=(Exception,))
# No trials single-objective study
optuna.create_study(study_name="single-no-trials", storage=storage, sampler=sampler)