Scalable_Pipeline_Article

Ordinal_Columns.py

class Ordinal_Columns(BaseEstimator, TransformerMixin):
    def __init__(self, names=['Deck','Deck2']): #,'Deck2'
        self.names = names

    def fit(self, X, y=None):
        return self

    def transform(self, X, y=None):
        if len(self.names) > 0:
            if len(self.names) == 1:
                X_ = X[self.names].copy()
            else:
                X_ = X[self.names].copy()
        return X_