Expose references to the PCA components and column means (#327)

rust-ml · Jan 13, 2024 · 0ade794 · 0ade794
1 parent d495562
commit 0ade794
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/algorithms/linfa-reduction/src/pca.rs b/algorithms/linfa-reduction/src/pca.rs
@@ -176,6 +176,16 @@ impl Pca<f64> {
         ex_var / sum_ex_var
     }
 
+    /// Return the components
+    pub fn components(&self) -> &Array2<f64> {
+        &self.embedding
+    }
+
+    /// Return the mean
+    pub fn mean(&self) -> &Array1<f64> {
+        &self.mean
+    }
+
     /// Return the singular values
     pub fn singular_values(&self) -> &Array1<f64> {
         &self.sigma