Source code for permetrics.utils.encoder

#!/usr/bin/env python
# Created by "Thieu" at 06:18, 26/07/2023 ----------%                                                                               
#       Email: nguyenthieu2102@gmail.com            %                                                    
#       Github: https://github.com/thieu1995        %                         
# --------------------------------------------------%

import numpy as np


[docs]class LabelEncoder: def __init__(self): self.classes_ = None self.encoded_classes_ = None
[docs] def fit(self, y): self.classes_, indices = np.unique(y, return_inverse=True) self.encoded_classes_ = np.arange(len(self.classes_)) return self
[docs] def transform(self, y): return np.searchsorted(self.classes_, y)
[docs] def inverse_transform(self, y): return self.classes_[y]
[docs] def fit_transform(self, y): self.fit(y) return self.transform(y)