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

Fix importance cache

Browse Source
This commit is contained in:
c-bata
2022-03-20 01:37:01 +09:00
parent 1f699f2bd7
commit b415eaf6d6
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
optuna_dashboard/_importance.py
+3 -1
View File
@@ -60,7 +60,9 @@ def get_param_importance_from_trials_cache(
cache_key = f"{study_id}:{objective_id}"
with param_importance_cache_lock:
cache_n_trial, cache_importance = param_importance_cache.get(cache_key, [0, {}])
cache_n_trial, cache_importance = param_importance_cache.get(
cache_key, [0, {"target_name": target_name, "param_importances": []}]
)
if n_completed_trials == cache_n_trial:
return cache_importance