diff --git a/optuna_dashboard/_app.py b/optuna_dashboard/_app.py index 1c9c0119..5057353d 100644 --- a/optuna_dashboard/_app.py +++ b/optuna_dashboard/_app.py @@ -271,6 +271,14 @@ def create_app( ) if plot_type == "contour": fig = optuna.visualization.plot_contour(study) + elif plot_type == "slice": + fig = optuna.visualization.plot_slice(study) + elif plot_type == "parallel_coordinate": + fig = optuna.visualization.plot_parallel_coordinate(study) + elif plot_type == "rank": + fig = optuna.visualization.plot_rank(study) + elif plot_type == "edf": + fig = optuna.visualization.plot_edf(study) else: response.status = 404 # Not found return {"reason": f"plot_type={plot_type} is not supported."} diff --git a/optuna_dashboard/ts/apiClient.ts b/optuna_dashboard/ts/apiClient.ts index 3798f602..46fe3d0e 100644 --- a/optuna_dashboard/ts/apiClient.ts +++ b/optuna_dashboard/ts/apiClient.ts @@ -448,6 +448,10 @@ type PlotResponse = { } export enum PlotType { Contour = "contour", + Slice = "slice", + ParallelCoordinate = "parallel_coordinate", + Rank = "rank", + EDF = "edf", } export const getPlotAPI = ( studyId: number,