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 vote303 views1 answer
Creating a bar plot bar | creating a bar chart
Python
26 votes310 views5 answers
Display max rows pandas
Python
0 vote317 views1 answer
@foreach 1 numper
Python
1 vote270 views1 answer
Wonsan
Python
23 votes372 views3 answers
Install django rest framework
Python Django
4 votes279 views3 answers
Python remove everything after character
Python
1 vote328 views4 answers
How to do a print statement in python
Python
22 votes288 views3 answers
Python except error as e
Python
0 vote328 views1 answer
How to read comment before the root element of xml python
Python
0 vote301 views1 answer
Python list key string
Python