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 vote214 views1 answer
Spark group by alias
Python
0 vote209 views1 answer
How to get the highest number of an input in python
Python
67 votes185 views6 answers
Activate venv
Python
1 vote192 views1 answer
Python read entire file as string
Python
4 votes228 views2 answers
Django composite primary key
Python Django
0 vote208 views1 answer
Pandas column by index
Python
2 votes219 views1 answer
Importying listviewin django
Python Django
0 vote203 views1 answer
Api.models.DoesNotExist: matching query does not exist
Python
0 vote196 views1 answer
Pandas and operator
Python
1 vote203 views7 answers
Java to python
Python