Label binarizer

>>> from sklearn import preprocessing
>>> lb = preprocessing.LabelBinarizer()
>>> lb.fit([1, 2, 6, 4, 2])
LabelBinarizer()
>>> lb.classes_
array([1, 2, 4, 6])
>>> lb.transform([1, 6])
array([[1, 0, 0, 0],
       [0, 0, 0, 1]])

Sklearn labelbinarizer in pipeline

from sklearn.preprocessing import LabelBinarizer

class LabelBinarizerPipelineFriendly(LabelBinarizer):
    def fit(self, X, y=None):
        """this would allow us to fit the model based on the X input."""
        super(LabelBinarizerPipelineFriendly,self).fit(X)
    def transform(self, X, y=None):
        return super(LabelBinarizerPipelineFriendly, self).transform(X)
    def fit_transform(self, X, y=None):
        return super(LabelBinarizerPipelineFriendly, self).fit(X).transform(X)

Source: github.com

Similar codes

0 vote175 views1 answer
AttributeError: module 'tensorflow._api.v2.train' has no attribute 'AdamOptiimizer'
Python
128 votes241 views6 answers
Installing pip
Python
0 vote193 views1 answer
Does np.random.randint have a seed
Python
0 vote326 views2 answers
Pytorch mse mae
Python
0 vote189 views1 answer
Change marker border color plotly
Python
-2 votes203 views6 answers
Django download
Python
1 vote196 views2 answers
Pygame collisions
Python
0 vote185 views2 answers
Dict to bytes python
Python
11 votes175 views2 answers
List map lambda python
Python
0 vote231 views1 answer
Python max value of list of tuples
Python