Update visualization.py

visualization.py

@@ -219,19 +219,6 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
     return fig
 
 
-# Function to create the correlation heatmap
-def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
-    data = np.vstack((mse_embeddings, mse_posture, mse_voice)).T
-    df = pd.DataFrame(data, columns=["Facial Features", "Body Posture", "Voice"])
-    corr = df.corr()
-
-    plt.figure(figsize=(10, 8), dpi=300)
-
-    heatmap = sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
-    plt.title('Correlation Heatmap of MSEs')
-    plt.tight_layout()
-    return plt.gcf()
-
 def plot_stacked_mse_heatmaps(mse_face, mse_posture, mse_voice, df, title="Combined MSE Heatmaps"):
     plt.figure(figsize=(20, 6), dpi=300)
     fig, (ax1, ax2, ax3) = plt.subplots(3, 1, figsize=(20, 8), sharex=True, gridspec_kw={'height_ratios': [1, 1, 1.2], 'hspace': 0})