Updating results

python/logistic/src/report_utils.py

@@ -16,7 +16,8 @@ def plot_pdf(
     cov_pars,
     fontsize = {"ticks":16, "axes_labels":16, "plot_labels": 18},
     width = 6,
-    height = 4
+    height = 4,
+    pars_labels = ["beta0", "beta1"],
 ):
     """
     Plot joint KDEs for two 2D datasets.
@@ -59,7 +60,7 @@ def plot_pdf(
     # FIRST
     g1 = sns.jointplot(x=data1[:, 0], y=data1[:, 1], kind="kde", fill=True)
     g1.ax_joint.tick_params(labelsize=fontsize["ticks"])
-    g1.set_axis_labels(xlabel=r"$\beta_0$", ylabel=r"$\beta_1$", fontsize=fontsize["axes_labels"])
+    g1.set_axis_labels(xlabel=pars_labels[0], ylabel=pars_labels[1], fontsize=fontsize["axes_labels"])
     g1.figure.text(0.04, 0.9, "(a)", fontsize=fontsize["plot_labels"], ha="center", va="center")
     
     g1.figure.savefig(os.path.join(results_path,  file1), bbox_inches="tight")
@@ -68,7 +69,7 @@ def plot_pdf(
     # SECOND
     g2 = sns.jointplot(x=data2[:, 0], y=data2[:, 1], kind="kde", fill=True)
     g2.ax_joint.tick_params(labelsize=fontsize["ticks"])
-    g2.set_axis_labels(xlabel=r"$\beta_0$", ylabel=r"$\beta_1$", fontsize=fontsize["axes_labels"])
+    g2.set_axis_labels(xlabel=pars_labels[0], ylabel=pars_labels[1], fontsize=fontsize["axes_labels"])
     g2.figure.text(0.04, 0.9, "(b)", fontsize=fontsize["plot_labels"], ha="center", va="center")
 
     # MLE point on the joint axes
@@ -633,6 +634,7 @@ def plot_from_dataframes_basic(
     jitter_y_full=None,
     jitter_y_trim=None,
     random_state=123,
+    plot_labs = None,
 ):
     """
     Plot raw data and fitted curves. One panel per scale.
@@ -670,8 +672,10 @@ def plot_from_dataframes_basic(
         squeeze=False,
     )
     axs = axs[0]
-
-    plot_labs = dict(zip(scales, list("ABCDEFGHIJKLMNOPQRSTUVWXYZ")[:len(scales)]))
+    
+    if plot_labs is None:
+        plot_labs = dict(zip(scales, list("ABCDEFGHIJKLMNOPQRSTUVWXYZ")[:len(scales)]))
+    
     if verbose:
         print(plot_labs)