add optuna plots to report (#336)

mljar · Apr 17, 2021 · bb760c1 · bb760c1
1 parent c36f892
commit bb760c1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/supervised/tuner/optuna/tuner.py b/supervised/tuner/optuna/tuner.py
@@ -312,6 +312,7 @@ def plot_study(self, algorithm, data_type, study):
                 with warnings.catch_warnings():
                     warnings.simplefilter("ignore")
                     plt.figure()
+                    plt.rcParams["axes.grid"] = title != "parallel_coordinate"
                     plot(study)
                     plt.tight_layout(pad=2.0)
                     fname = f"{key}_{title}.png"