Source code for autogluon.features.generators.identity
import logging
from pandas import DataFrame
from .abstract import AbstractFeatureGenerator
logger = logging.getLogger(__name__)
[docs]
class IdentityFeatureGenerator(AbstractFeatureGenerator):
"""IdentityFeatureGenerator simply passes the data along without alterations."""
def _fit_transform(self, X: DataFrame, **kwargs) -> (DataFrame, dict):
X_out = self._transform(X)
return X_out, self.feature_metadata_in.type_group_map_special
def _transform(self, X: DataFrame) -> DataFrame:
return X
@staticmethod
def get_default_infer_features_in_args() -> dict:
return dict()
def _more_tags(self):
return {"feature_interactions": False}