Update visualization.py

visualization.py

@@ -236,17 +236,17 @@ def plot_stacked_mse_heatmaps(mse_face, mse_posture, mse_voice, df, title="Combi
     fig, (ax1, ax2, ax3) = plt.subplots(3, 1, figsize=(20, 6), sharex=True, gridspec_kw={'height_ratios': [1, 1, 1.2], 'hspace': 0})
 
     # Face heatmap
-    sns.heatmap(mse_face.reshape(1, -1), cmap='YlOrRd', cbar=False, ax=ax1, xticklabels=False, yticklabels=False)
+    sns.heatmap(mse_face.reshape(1, -1), cmap='Reds', cbar=False, ax=ax1, xticklabels=False, yticklabels=False)
     ax1.set_ylabel('Face', rotation=0, ha='right', va='center')
     ax1.yaxis.set_label_coords(-0.01, 0.5)
 
     # Posture heatmap
-    sns.heatmap(mse_posture.reshape(1, -1), cmap='YlOrRd', cbar=False, ax=ax2, xticklabels=False, yticklabels=False)
+    sns.heatmap(mse_posture.reshape(1, -1), cmap='Reds', cbar=False, ax=ax2, xticklabels=False, yticklabels=False)
     ax2.set_ylabel('Posture', rotation=0, ha='right', va='center')
     ax2.yaxis.set_label_coords(-0.01, 0.5)
 
     # Voice heatmap
-    sns.heatmap(mse_voice.reshape(1, -1), cmap='YlOrRd', cbar=False, ax=ax3, yticklabels=False)
+    sns.heatmap(mse_voice.reshape(1, -1), cmap='Reds', cbar=False, ax=ax3, yticklabels=False)
     ax3.set_ylabel('Voice', rotation=0, ha='right', va='center')
     ax3.yaxis.set_label_coords(-0.01, 0.5)