Add tests

This commit is contained in:
porink0424
2024-05-10 13:06:45 +09:00
parent 7b4ecd417f
commit 0ec4f085f7
2 changed files with 24 additions and 1 deletions
+18
View File
@@ -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()
+6 -1
View File
@@ -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)
})
})