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 vote96 views3 answers
Isprime in python
Python
10 votes103 views3 answers
Python print traceback from exception
Python
6 votes104 views1 answer
Utf8 python encodage line
Python
0 vote69 views2 answers
Python fraction simplify
Python
0 vote103 views1 answer
Minmaxscaler transform
Python
9 votes124 views2 answers
TypeError: method() takes 1 positional argument but 2 were given
Python
0 vote134 views3 answers
Assert in python
Python
0 vote72 views2 answers
Tkinter disable resize
Python
29 votes128 views3 answers
'pipwin' is not recognized as an internal or external command
Python
0 vote119 views1 answer
Python Print Variable Using the f-string in the print statement
Python