Merge pull request #993 from porink0424/fix/categorical-choices-visua…

…lization-issue Fix plot visualization issues with categorical parameters
optuna · Nov 20, 2024 · dbb0862 · dbb0862
2 parents 29ff5aa + d9d64bd
commit dbb0862
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tslib/react/src/components/PlotParallelCoordinate.tsx b/tslib/react/src/components/PlotParallelCoordinate.tsx
@@ -173,7 +173,7 @@ const plotCoordinate = (
     if (s.distribution.type === "CategoricalDistribution") {
       // categorical
       const vocabArr: string[] = s.distribution.choices.map(
-        (c) => c?.toString() ?? "null"
+        (c) => c?.value ?? "null"
       )
       const tickvals: number[] = vocabArr.map((_, i) => i)
       return {

diff --git a/tslib/react/src/components/PlotSlice.tsx b/tslib/react/src/components/PlotSlice.tsx
@@ -268,7 +268,7 @@ const plotSlice = (
     }
   } else {
     const vocabArr = selectedParamSpace.distribution.choices.map(
-      (c) => c?.toString() ?? "null"
+      (c) => c?.value ?? "null"
     )
     const tickvals: number[] = vocabArr.map((_v, i) => i)
     layout.xaxis = {

diff --git a/tslib/react/src/components/TrialTable.tsx b/tslib/react/src/components/TrialTable.tsx
@@ -78,7 +78,7 @@ export const TrialTable: FC<{
       const sortable = s.distribution.type !== "CategoricalDistribution"
       const filterChoices: (string | null)[] | undefined =
         s.distribution.type === "CategoricalDistribution"
-          ? s.distribution.choices.map((c) => c?.toString() ?? "null")
+          ? s.distribution.choices.map((c) => c?.value ?? "null")
           : undefined
       const hasMissingValue = trials.some(
         (t) => !t.params.some((p) => p.name === s.name)