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 vote173 views1 answer
Compute Jordan normal form of matrix in Python / NumPy
Python
0 vote167 views1 answer
Python how to align text writen to a file
Python
4 votes181 views1 answer
Python request remove warning
Python
1 vote180 views1 answer
Markers seaborn
Python
1 vote162 views1 answer
The name tf.summary.merge_all is deprecated. Please use tf.compat.v1.summary.merge_all
Python
0 vote164 views1 answer
Comment arrêter un jeu en appuyant sur une touche python
Python
0 vote166 views1 answer
Get hwid python
Python
0 vote166 views2 answers
Arma-garch python
Python
37 votes127 views4 answers
Python pick random from list
Python
77 votes167 views6 answers
Reverse a string python
Python