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

2 votes180 views1 answer
Getting dummies for a column in pandas dataframe
Python
0 vote169 views1 answer
Values django
Python
1 vote177 views1 answer
Telnet via jump host using python
Python
0 vote216 views2 answers
Export keras model at specific epoch
Python
0 vote178 views1 answer
Find percentage of missing values in a column in python
Python
0 vote172 views1 answer
Sprints for fat loss
Python
0 vote240 views2 answers
How to download a .xlsx file from google colab
Python
0 vote169 views3 answers
Pandas reading each xlsx file in folder
Python
0 vote197 views5 answers
Pandas sum
Python
0 vote189 views9 answers
Odoo.py odoo 14
Python