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

1 vote355 views1 answer
Formula for compounding interest in python
Python
29 votes309 views2 answers
Insertion sort python
Python
21 votes336 views6 answers
Pandas convert column to datetime
Python
0 vote317 views1 answer
Check tensor type tensorflow
Python
0 vote340 views4 answers
AttributeError: 'Database' object has no attribute 'remove'
Python
0 vote324 views1 answer
Merge two arrays python with three one
Python
4 votes307 views1 answer
Np.random.uniform
Python
3 votes336 views2 answers
List vs tuple
Python
32 votes346 views4 answers
Python check if list contains value
Python
0 vote313 views1 answer
Find index corresponding to maximum value pandas
Python