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 vote168 views1 answer
Jinja macro import
Python
1 vote210 views1 answer
Pygame doesnt dedect collision between sprite and image
Python
0 vote184 views2 answers
Write a program to check whether a character is vowel or consonant in python
Python
0 vote182 views1 answer
Drop duplicates pandas considering lowercase
Python
0 vote177 views1 answer
HTTP Error 403: Forbidden django account signup email
Python
0 vote200 views1 answer
Rotate image by specific angle opencv
Python
1 vote178 views1 answer
Selenium close current driver but don't quit
Python
4 votes264 views1 answer
Finding a substring between 2 strings
Python
0 vote174 views1 answer
Python pretty print list of tuples
Python
1 vote170 views1 answer
Image delete in django from the folder
Python