Source code for pyreal.transformers.type_cast

from pyreal.transformers import TransformerBase


class BoolToIntCaster(TransformerBase):
    def data_transform(self, x):
        """
        Transform booleans to integers
        Args:
            x (DataFrame of shape (n_instances, n_features)):
                The dataset to transform

        Returns:
            DataFrame of shape (n_instances, n_transformed_features):
                The transformed dataset
        """
        return x.replace({False: 0, True: 1})