#7; add ValueMapper in src.preprocess

abhi8893 · Jul 2, 2021 · fde7099 · fde7099
1 parent 7175b3d
commit fde7099
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/mlops/mpg-pred-end-to-end-ml/src/preprocess.py b/mlops/mpg-pred-end-to-end-ml/src/preprocess.py
@@ -0,0 +1,16 @@
+
+from sklearn.base import BaseEstimator, TransformerMixin 
+import pandas as pd
+
+class ValueMapper(BaseEstimator, TransformerMixin):
+
+    def __init__(self, mapper: dict):
+        self.mapper = mapper
+
+    def fit(self, X, y=None):
+        # TODO: Get column wise default mappings
+        return self
+
+    def transform(self, X):
+        X = X.apply(lambda col: col.map(self.mapper)).to_numpy()
+        return X