feat: statistics tests

2025-09-28 17:16:02 +02:00
parent ef4ec70fcc
commit 9039e9a4cf
20 changed files with 33006 additions and 1140 deletions
@@ -0,0 +1,15 @@
+import seaborn as sns
+
+from research.statistics.utils import LETTERS
+
+
+def plot_transition_matrix(ax, df_probs, title=""):
+    hm = sns.heatmap(
+        df_probs.loc[list(LETTERS), list(LETTERS)],
+        cmap="Reds",
+        annot=False,
+        cbar=False,
+        ax=ax
+    )
+    ax.set_title(title, fontsize=12)
+    return hm