Update visualization.py

visualization.py

@@ -5,6 +5,7 @@ import pandas as pd
 from matplotlib.patches import Rectangle
 from utils import seconds_to_timecode
 from anomaly_detection import determine_anomalies
+plt.style.use('seaborn')
 
 def plot_mse(df, mse_values, title, color='navy', time_threshold=3, anomaly_threshold=4):
     plt.figure(figsize=(16, 8), dpi=400)
@@ -198,4 +199,24 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
 
     plt.tight_layout()
     plt.close()
-    return fig
+    return fig
+
+def create_kdeplot(df, mse_embeddings, mse_posture):
+    plt.figure(figsize=(10, 6), dpi=300)
+    df['MSE_Embeddings'] = mse_embeddings
+    df['MSE_Posture'] = mse_posture
+    sns.kdeplot(data=df, x='MSE_Embeddings', y='MSE_Posture', cmap="YlOrRd", shade=True, cbar=True)
+    plt.xlabel('Facial Features MSE')
+    plt.ylabel('Body Posture MSE')
+    plt.title('KDE Plot of Facial Features vs Body Posture MSE')
+    plt.close()
+    return plt.gcf()
+
+def create_jointplot(df, mse_embeddings, mse_posture):
+    plt.figure(figsize=(10, 6), dpi=300)
+    df['MSE_Embeddings'] = mse_embeddings
+    df['MSE_Posture'] = mse_posture
+    joint_plot = sns.jointplot(data=df, x='MSE_Embeddings', y='MSE_Posture', kind='hex', cmap="YlOrRd")
+    joint_plot.fig.suptitle('Joint Plot of Facial Features vs Body Posture MSE', y=1.02)
+    plt.close()
+    return joint_plot.fig