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 vote247 views1 answer
How to openn file dialog in tkinter
Python
1 vote279 views1 answer
Jupyter notebook delete the output
Python
0 vote274 views6 answers
Python django
Python
0 vote252 views1 answer
Phyton 2.7 convert timedelta to string
Python
0 vote271 views1 answer
How to make a minute counter in python
Python
0 vote253 views1 answer
Re mobile no validate python
Python
1 vote252 views1 answer
Pycharm rename file
Python
26 votes359 views6 answers
Flask run
Python Flask
0 vote256 views1 answer
Create a dict from variables and give name
Python
0 vote258 views1 answer
How to deal with SettingWithCopyWarning in Pandas
Python