DOC Fix FutureWarning in plot_successive_halving_heatmap.html (#24968)

ArturoAmorQ · glemaitre · web-flow · commit 43550f011a3a · 2022-11-17T15:41:51.000+01:00
Co-authored-by: Guillaume Lemaitre &lt;g.lemaitre58@gmail.com&gt;
diff --git a/examples/model_selection/plot_successive_halving_heatmap.py b/examples/model_selection/plot_successive_halving_heatmap.py
@@ -54,8 +54,10 @@
 
 def make_heatmap(ax, gs, is_sh=False, make_cbar=False):
     """Helper to make a heatmap."""
-    results = pd.DataFrame.from_dict(gs.cv_results_)
-    results["params_str"] = results.params.apply(str)
+    results = pd.DataFrame(gs.cv_results_)
+    results[["param_C", "param_gamma"]] = results[["param_C", "param_gamma"]].astype(
+        np.float64
+    )
     if is_sh:
         # SH dataframe: get mean_test_score values for the highest iter
         scores_matrix = results.sort_values("iter").pivot_table(
