Update visualization.py
Browse files- visualization.py +0 -18
visualization.py
CHANGED
@@ -268,21 +268,3 @@ def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
|
|
268 |
plt.title('Correlation Heatmap of MSEs')
|
269 |
plt.tight_layout()
|
270 |
return plt.gcf()
|
271 |
-
|
272 |
-
def plot_3d_scatter(mse_embeddings, mse_posture, mse_voice):
|
273 |
-
fig = plt.figure()
|
274 |
-
plt.figure(figsize=(16, 8), dpi=300)
|
275 |
-
ax = fig.add_subplot(111, projection='3d')
|
276 |
-
|
277 |
-
# Scatter plot
|
278 |
-
ax.scatter(mse_posture, mse_embeddings, mse_voice, c=['purple']*len(mse_posture), label='Body Posture', alpha=0.6)
|
279 |
-
ax.scatter(mse_posture, mse_embeddings, mse_voice, c=['navy']*len(mse_embeddings), label='Facial Features', alpha=0.6)
|
280 |
-
ax.scatter(mse_posture, mse_embeddings, mse_voice, c=['green']*len(mse_voice), label='Voice', alpha=0.6)
|
281 |
-
|
282 |
-
ax.set_xlabel('Body Posture MSE')
|
283 |
-
ax.set_ylabel('Facial Features MSE')
|
284 |
-
ax.set_zlabel('Voice MSE')
|
285 |
-
|
286 |
-
ax.legend()
|
287 |
-
plt.close(fig)
|
288 |
-
return fig
|
|
|
268 |
plt.title('Correlation Heatmap of MSEs')
|
269 |
plt.tight_layout()
|
270 |
return plt.gcf()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|