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 vote151 views1 answer
Keras functional api embedding layer
Python
8 votes163 views3 answers
How to set default python version in macos
Python
0 vote166 views1 answer
Python os.getenv not working
Python
4 votes202 views2 answers
Arrayfield django example
Python Django
4 votes172 views3 answers
Astype float across columns pandas
Python
1 vote165 views1 answer
Pandas read_csv random rows
Python
0 vote163 views3 answers
Tower of Hanoi
Python
0 vote132 views1 answer
Pysimplegui get value from textbox
Python
0 vote156 views1 answer
Proj 4.9.0 must be installed.
Python
0 vote149 views1 answer
Explain the use of return keyword python
Python