From 0b6504a1722c4fb4a1b6cebe640c2ad20b7389ae Mon Sep 17 00:00:00 2001 From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net> Date: Mon, 10 Mar 2025 14:32:12 +0000 Subject: [PATCH] Edit reduction_dimension_2d.py --- utils/reduction_dimension_2d.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/utils/reduction_dimension_2d.py b/utils/reduction_dimension_2d.py index e69de29..ce6d9c7 100644 --- a/utils/reduction_dimension_2d.py +++ b/utils/reduction_dimension_2d.py @@ -0,0 +1,8 @@ +from sklearn.decomposition import PCA + + + +def reduce_embeddings_2d(embeddings, n_components=2, random_state=42): + """Réduit les embeddings à 2 dimensions avec PCA.""" + pca = PCA(n_components=n_components, random_state=random_state) + return pca.fit_transform(embeddings) \ No newline at end of file -- GitLab