Add NCA

toandaominh1997 · Mar 9, 2021 · 359c9ff · 359c9ff
1 parent 6db71c0
commit 359c9ff
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/pipelineservice/neighbors/__init__.py b/pipelineservice/neighbors/__init__.py
@@ -0,0 +1 @@
+from ._neighbors import neighborhoodComponentsAnalysis
diff --git a/pipelineservice/neighbors/_neighbors.py b/pipelineservice/neighbors/_neighbors.py
@@ -0,0 +1,12 @@
+import numpy as np
+import pandas as pd
+from sklearn.neighbors import NeighborhoodComponentsAnalysis
+
+
+class neighborhoodComponentsAnalysis(NeighborhoodComponentsAnalysis):
+    feature_name = 'nca'
+    def transform(self, X):
+        data = super().transform(X)
+        cols = [f'{self.feature_name}_{i}' for i in range(data.shape[1])]
+        return pd.DataFrame(data, columns = cols, index = X.index)
+