/Python/Sklearn labelbinarizer in pipeline

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 vote198 views1 answer
How to get a list of all variables in memory python
Python
0 vote216 views2 answers
Python if not none in one line
Python
0 vote238 views1 answer
Python string: .lower()
Python
2 votes211 views1 answer
Making lists with loops in one line python
Python
2 votes208 views1 answer
How to start ftpd server with python
Python
1 vote223 views1 answer
How to get sum specific columns value in machine learning
Python
0 vote212 views1 answer
If loop in python syntax
Python
0 vote218 views1 answer
Python calculated row in dataframe subtract
Python
0 vote197 views1 answer
Add key if not exists python
Python
6 votes247 views6 answers
How to check if a value is nan in python
Python