Method __init__

numpy_ml/preprocessing/general.py:55–67

Convert between category labels and their one-hot vector representations. Parameters ---------- categories : list of length `C` List of the unique category labels for the items to encode.

(self)

class OneHotEncoder:
    def __init__(self):
        """
        Convert between category labels and their one-hot vector
        representations.

        Parameters
        ----------
        categories : list of length `C`
            List of the unique category labels for the items to encode.
        """
        self._is_fit = False
        self.hyperparameters = {}
        self.parameters = {"categories": None}

    def __call__(self, labels):
        return self.transform(labels)