Add OrdinalEncoder

toandaominh1997 · Aug 3, 2021 · 2946fc6 · 2946fc6
1 parent c792b80
commit 2946fc6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pipelineservice/preprocessing/_data.py b/pipelineservice/preprocessing/_data.py
@@ -2,7 +2,7 @@
 import pandas as pd
 from sklearn.base import BaseEstimator, TransformerMixin
 from sklearn.preprocessing import PolynomialFeatures, QuantileTransformer, PowerTransformer, KBinsDiscretizer, KernelCenterer
-from sklearn.preprocessing import StandardScaler, RobustScaler, LabelEncoder, MinMaxScaler
+from sklearn.preprocessing import StandardScaler, RobustScaler, LabelEncoder, MinMaxScaler, OrdinalEncoder
 
 class polynomialFeatures(PolynomialFeatures):
     def transform(self, X):
@@ -35,6 +35,10 @@ def transform(self, X):
         data = super().transform(X)
         return pd.DataFrame(data, columns = X.columns, index = X.index)
 
+class ordinalEncoder(OrdinalEncoder):
+    def transform(self, X):
+        data = super().transform(X)
+        return pd.DataFrame(data, columns = X.columns, index = X.index)
 class labelEncoder(LabelEncoder):
     def transform(self, X):
         data = super().transform(X)