Scalable_Pipeline_Article

CatFeatureEngineerTransformer.py

class CatFeatureEngineerTransformer(BaseEstimator, TransformerMixin):
    def __init__(self, home_dest = True):
        self.home_dest = home_dest
        
    def fit(self, X, y=None):
        return self
    
    def transform(self, X, y=None):
        if self.home_dest:
            X['HomeDest'] = X.HomePlanet + ':' + X.Destination
            X.drop(columns=['HomePlanet', 'Destination'], inplace=True)
        return X