Set include_best_trial=True for v3 release

This commit is contained in:
c-bata
2022-03-24 12:34:58 +09:00
parent 22f5c9cf91
commit ba96fd624b
2 changed files with 9 additions and 5 deletions
+8 -5
View File
@@ -30,6 +30,8 @@ from optuna.storages import RedisStorage
from optuna.study import StudyDirection
from optuna.study import StudySummary
from optuna.trial import FrozenTrial
from optuna.version import __version__ as optuna_ver
from packaging import version
from . import _note as note
from ._importance import get_param_importance_from_trials_cache
@@ -121,12 +123,13 @@ def create_app(storage: BaseStorage, debug: bool = False) -> Bottle:
@app.get("/api/studies")
@json_api_view
def list_study_summaries() -> BottleViewReturn:
summaries = [
serialize_study_summary(summary)
for summary in storage.get_all_study_summaries()
]
if version.parse(optuna_ver) >= version.Version("3.0.0b0"):
summaries = storage.get_all_study_summaries(include_best_trial=True)
else:
summaries = storage.get_all_study_summaries()
serialized = [serialize_study_summary(summary) for summary in summaries]
return {
"study_summaries": summaries,
"study_summaries": serialized,
}
@app.post("/api/studies")
+1
View File
@@ -29,6 +29,7 @@ include_package_data = False
packages = find:
install_requires =
optuna>=2.4
packaging
bottle
typing-extensions;python_version<'3.8'
scikit-learn