From dd8c1793cfa7efee340880085713d7d5161a6647 Mon Sep 17 00:00:00 2001
From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net>
Date: Mon, 10 Mar 2025 14:38:04 +0000
Subject: [PATCH] Add new file
---
results/visualize_clusters.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
create mode 100644 results/visualize_clusters.py
diff --git a/results/visualize_clusters.py b/results/visualize_clusters.py
new file mode 100644
index 0000000..92dd491
--- /dev/null
+++ b/results/visualize_clusters.py
@@ -0,0 +1,13 @@
+import matplotlib.pyplot as plt
+
+def visualize_clusters(embeddings_2d, cluster_labels, title="Clusters visualisés en 2D"):
+ """Affiche les clusters en 2D avec un scatter plot."""
+ plt.figure(figsize=(10, 8))
+ plt.scatter(embeddings_2d[:, 0], embeddings_2d[:, 1], c=cluster_labels, cmap='viridis', alpha=0.7)
+ plt.title(title)
+ plt.xlabel('Dimension 1')
+ plt.ylabel('Dimension 2')
+ plt.colorbar(label='Cluster')
+ plt.grid(True)
+ plt.show()
+
--
GitLab