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 vote238 views1 answer
How to append to every second item in list python
Python
0 vote177 views2 answers
Nameerror name 'defaultdict' is not defined
Python
1 vote165 views1 answer
Python change base function
Python
0 vote149 views1 answer
Python boxplot legend
Python
13 votes142 views5 answers
Python get filename without extension
Python
48 votes149 views6 answers
Python read file line by line
Python
0 vote170 views1 answer
Destructuring in for loops python
Python
1 vote158 views1 answer
Import antigravity in python
Python
0 vote184 views2 answers
Python unsupported operand type(s) for & 'str' and 'str'
Python
86 votes159 views4 answers
How to delete a file in python
Python