Source code for autoflow.pipeline.components.preprocessing.encode.base

import pandas as pd

from autoflow.pipeline.components.feature_engineer_base import AutoFlowFeatureEngineerAlgorithm


[docs]class BaseEncoder(AutoFlowFeatureEngineerAlgorithm):

    def _transform_proc(self, X):
        if X is None:
            return None
        else:
            X_ = X.astype(str)
            trans = self.estimator.transform(X_)
            trans[pd.DataFrame(X) == -999] = -999  # todo: 有没有更优化的解决办法
            return trans

[docs]    def core_fit(self, estimator, X, y, X_valid=None, y_valid=None, X_test=None,
                 y_test=None, feature_groups=None, columns_metadata=None):
        X_ = X.astype(str)
        return estimator.fit(X_, y)