Update gpa.py

MaxHalford · Dec 16, 2023 · 10fbd77 · 10fbd77
1 parent d3e34f1
commit 10fbd77
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/prince/gpa.py b/prince/gpa.py
@@ -14,10 +14,9 @@ class GPA(base.BaseEstimator, base.TransformerMixin):
     """Generalized Procrustes Analysis (GPA).
 
     Algorithm outline:
-        (https://wikipedia.org/wiki/Generalized_Procrustes_analysis)
+    
     1. Choose a reference shape.
-    2. Apply Procrustes Analysis to superimpose all shapes to the reference
-        shape.
+    2. Apply Procrustes Analysis to superimpose all shapes to the reference shape.
     3. Compute the mean shape of the superimposed shapes.
     4. Repeat steps 2 and 3 until convergence.
 
@@ -43,6 +42,11 @@ class GPA(base.BaseEstimator, base.TransformerMixin):
     random_state
         Determines random number generation for initialization when `init=='random'`.
 
+    References
+    ----------
+    https://wikipedia.org/wiki/Generalized_Procrustes_analysis
+    https://medium.com/@olga_kravchenko/generalized-procrustes-analysis-with-python-numpy-c571e8e8a421
+
     """
 
     def __init__(