Create NumericSelector.py

NumericSelector.py

@@ -0,0 +1,11 @@
+from sklearn.base import BaseEstimator, TransformerMixin
+
+class NumericSelector(BaseEstimator, TransformerMixin):
+
+    def fit(self, X, y=None):
+        self.numeric_cols_ = X.select_dtypes(include=[float, int]).columns
+        return self
+
+    def transform(self, X):
+        X_num = X[self.numeric_cols_]
+        return X_num