Update visualization.py
Browse files- 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
|
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))
|