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

Fix target arg for importance

Browse Source
This commit is contained in:
c-bata committed 2023-01-06 10:14:52 +09:00
1 parent cac89cc9ef
commit 7f0f7bb139
1 file changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files
optuna_dashboard/_importance.py
+3 -1
View File
@@ -96,7 +96,9 @@ def get_param_importance_from_trials_cache(
return cache_importance
study = StudyWrapper(storage, study_id, trials)
importance = _get_param_importances(study, completed_trials)
importance = _get_param_importances(
study, completed_trials, target=lambda t: t.values[objective_id]
)
converted = convert_to_importance_type(importance, trials)
param_importance_cache[cache_key] = (n_completed_trials, converted)
return converted