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

-3 votes146 views5 answers
Multiplication table in python
Python
0 vote120 views1 answer
Select features and label from df
Python
0 vote125 views1 answer
Computercraft turtle place block
Python
0 vote132 views1 answer
How to check if string is in byte formate pythin
Python
68 votes129 views6 answers
Create directory in python
Python
0 vote137 views1 answer
Run python with options
Python
0 vote121 views1 answer
Replace all the empty rows in the column with the value that you have identified
Python
0 vote134 views1 answer
Seaborn documentation x axis range
Python
21 votes132 views2 answers
Python extract substring
Python
23 votes134 views3 answers
Django drf
Python