reab5555 commited on
Commit
2bfe741
·
verified ·
1 Parent(s): 338dc52

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +1 -1
visualization.py CHANGED
@@ -154,7 +154,7 @@ def plot_combined_mse(df, mse_embeddings, mse_posture, mse_voice, title):
154
  plt.close()
155
  return fig
156
 
157
- def plot_combined_heatmap(mse_embeddings, mse_posture, mse_voice, df):
158
  # Normalize MSE values
159
  mse_embeddings_norm = (mse_embeddings - np.min(mse_embeddings)) / (np.max(mse_embeddings) - np.min(mse_embeddings))
160
  mse_posture_norm = (mse_posture - np.min(mse_posture)) / (np.max(mse_posture) - np.min(mse_posture))
 
154
  plt.close()
155
  return fig
156
 
157
+ def combined_heatmap(mse_embeddings, mse_posture, mse_voice, df):
158
  # Normalize MSE values
159
  mse_embeddings_norm = (mse_embeddings - np.min(mse_embeddings)) / (np.max(mse_embeddings) - np.min(mse_embeddings))
160
  mse_posture_norm = (mse_posture - np.min(mse_posture)) / (np.max(mse_posture) - np.min(mse_posture))