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

2 votes358 views2 answers
Django exclude queryset
Python
3 votes358 views1 answer
2x + 9y = -7 6x - 3y = 9
Python
0 vote283 views1 answer
Dip programming language
Python
103 votes375 views6 answers
Install opencv python
Python
0 vote334 views1 answer
Scipy.signal lfilter
Python
22 votes316 views6 answers
Sort key python
Python
9 votes445 views5 answers
Proxy request python
Python
10 votes339 views4 answers
Discord.py fetch channel
Python
52 votes345 views2 answers
Index in list
Python
4 votes364 views2 answers
Compare two dictionaries in python
Python