mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-09-10 12:23:22 +08:00
Add tests
This commit is contained in:
@@ -78,6 +78,24 @@ def create_optuna_storage(storage: BaseStorage) -> None:
|
||||
|
||||
study.optimize(objective_single_nan_report, n_trials=100)
|
||||
|
||||
# Multi-objective study with metric names
|
||||
study = optuna.create_study(
|
||||
study_name="multi-objective-metric-names",
|
||||
storage=storage,
|
||||
directions=["minimize", "minimize"],
|
||||
)
|
||||
print(f"Generating {study.study_name} for {type(storage).__name__}...")
|
||||
study.set_metric_names(["value1", "value2"])
|
||||
|
||||
def objective_multi(trial: optuna.Trial) -> tuple[float, float]:
|
||||
x = trial.suggest_float("x", 0, 5)
|
||||
y = trial.suggest_float("y", 0, 3)
|
||||
v0 = 4 * x**2 + 4 * y**2
|
||||
v1 = (x - 5) ** 2 + (y - 5) ** 2
|
||||
return v0, v1
|
||||
|
||||
study.optimize(objective_multi, n_trials=50)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
remove_assets()
|
||||
|
||||
@@ -52,8 +52,13 @@ describe("Test Journal File Storage", async () => {
|
||||
)
|
||||
})
|
||||
|
||||
it("Check metric_names function", () => {
|
||||
const study = studies.find((s) => s.name === "multi-objective-metric-names")
|
||||
assert.deepStrictEqual(study.metric_names, ["value1", "value2"])
|
||||
})
|
||||
|
||||
it("Check the number of studies", () => {
|
||||
const N_STUDIES = 4
|
||||
const N_STUDIES = 5
|
||||
assert.strictEqual(studies.length, N_STUDIES)
|
||||
})
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user