Merge pull request optuna#5892 from fusawa-yugo/fusawa-yugo/check-visualization

Fix contour plot of matplotlib

Author: HideakiImamura

optuna/visualization/matplotlib/_contour.py

@@ -175,7 +175,12 @@ def _calculate_axis_data(
     returned_values: Sequence[int | float]
     if axis.is_cat:
         enc = _LabelEncoder()
-        returned_values = enc.fit_transform(list(map(str, values)))
+        # Fit LabelEncoder with all the categories in categorical distribution.
+        enc.fit(list(map(str, filter(lambda value: value is not None, axis.values))))
+        # Then transform the values using the fitted label encoder.
+        # Note that `values` may not include all the categories,
+        # so we use `axis.values` for fitting.
+        returned_values = enc.transform(list(map(str, values)))
         cat_param_labels = enc.get_labels()
         cat_param_pos = enc.get_indices()
     else: