Skip to content

Commit 56f6693

Browse files
authored
Merge pull request scikit-learn-contrib#21 from MaxBenChrist/fixGridSearch
make BorutaPy a sklearn Transformer
2 parents eb28240 + 647c192 commit 56f6693

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

boruta/boruta_py.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,10 @@
1212
import numpy as np
1313
import scipy as sp
1414
from sklearn.utils import check_random_state, check_X_y
15+
from sklearn.base import TransformerMixin, BaseEstimator
1516

16-
class BorutaPy(object):
17+
18+
class BorutaPy(BaseEstimator, TransformerMixin):
1719
"""
1820
Improved Python implementation of the Boruta R package.
1921

0 commit comments

Comments
 (0)