Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
|
@@ -309,4 +309,4 @@ def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
|
|
| 309 |
heatmap = sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
|
| 310 |
plt.title('Correlation Heatmap of MSEs')
|
| 311 |
plt.tight_layout()
|
| 312 |
-
return plt.gcf()
|
|
|
|
| 309 |
heatmap = sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
|
| 310 |
plt.title('Correlation Heatmap of MSEs')
|
| 311 |
plt.tight_layout()
|
| 312 |
+
return plt.gcf()
|