Update domain_space.py

domain_space.py

@@ -197,8 +197,8 @@ def make_domain_plot(df_synth_pred, explored_domain_space, x_list):
     """
     Create a plot with the uncertainty space and the training space
     """
-    uncertainty_min = df_synth_pred["uncertainty"].min()
-    uncertainty_max = df_synth_pred["uncertainty"].max()
+    uncertainty_min = df_synth_pred["y_pred"].min()
+    uncertainty_max = df_synth_pred["y_pred"].max()
 
     # df_synth_pred2, filter_cols = filter_dataframe(x_list[:6], df_synth_pred)
     df_synth_pred2, filter_cols = interpolate_all(x_list[:6], df_synth_pred)
@@ -210,9 +210,9 @@ def make_domain_plot(df_synth_pred, explored_domain_space, x_list):
         x=x_col,
         y=y_col,
         z=z_col,
-        color="uncertainty",
+        color="y_pred",
         range_color=[uncertainty_min, uncertainty_max],
-        hover_data={"uncertainty": ":.3f"},
+        hover_data={"y_pred": ":.3f"},
     )
 
     # Filter domain space