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

15 votes221 views6 answers
How to append list in python
Python
0 vote235 views1 answer
Sorts this RDD, which is assumed to consist of (key, value) pairs
Python
8 votes225 views5 answers
No module psycopg2
Python
30 votes227 views3 answers
How to replace null values in pandas
Python
4 votes237 views1 answer
Add x axis label python
Python
30 votes224 views6 answers
Django static files
Python
0 vote230 views1 answer
Running code once in a while loop python
Python
15 votes238 views5 answers
List of dataframe to dataframe
Python
0 vote237 views1 answer
Python sh command
Python
5 votes230 views3 answers
Convert list to dataframe
Python