reab5555 commited on
Commit
171299c
·
verified ·
1 Parent(s): 783453f

Update visualization.py

Browse files
Files changed (1) hide show
  1. 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()