[WIP] Update

for-ai · Oct 13, 2024 · 0ab3aaa · 0ab3aaa
1 parent 12d6492
commit 0ab3aaa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/analysis/avg_agreement_final.py b/analysis/avg_agreement_final.py
@@ -3,6 +3,7 @@
 import numpy as np
 
 FONT_SIZES = {"small": 12, "medium": 16, "large": 18}
+COLORS = {"green": "#355145", "purple": "#d8a6e5", "orange": "#fe7759"}
 
 PLOT_PARAMS = {
     "font.family": "serif",
@@ -48,7 +49,13 @@
 plt.figure(figsize=(7, 7))
 x_pos_sorted = np.arange(len(labels_sorted))
 
-ax1 = sns.barplot(x=x_pos_sorted, y=means_sorted, errorbar=None, color="gray", edgecolor="k")
+ax1 = sns.barplot(
+    x=x_pos_sorted,
+    y=means_sorted,
+    errorbar=None,
+    color=COLORS.get("orange"),
+    edgecolor=COLORS.get("green"),
+)
 plt.errorbar(x_pos_sorted, means_sorted, yerr=std_devs_sorted, fmt="none", c="black", capsize=5)
 
 # ax1.spines["top"].set_color("black")

diff --git a/analysis/plot_results.py b/analysis/plot_results.py
@@ -307,8 +307,8 @@ def plot_ling_dims(
             y=dim,
             data=lingdf,
             ax=ax,
-            color="gray",
-            edgecolor="k",
+            color=COLORS.get("orange"),
+            edgecolor=COLORS.get("green"),
             width=0.4 if dim == "Resource Availability" else 0.7,
         )
         ax.set_title(dim)