Skip to content

Commit ac23770

Browse files
authored
return importance history from fit (#88)
1 parent be7b270 commit ac23770

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

boruta/boruta_py.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,10 @@ class BorutaPy(BaseEstimator, TransformerMixin):
136136
best) features are assigned rank 1 and tentative features are assigned
137137
rank 2.
138138
139+
importance_history_ : array-like, shape [n_features, n_iters]
140+
141+
The calculated importance values for each feature across all iterations.
142+
139143
Examples
140144
--------
141145
@@ -378,6 +382,8 @@ def _fit(self, X, y):
378382
# all are selected, thus we set feature supports to True
379383
self.support_ = np.ones(n_feat, dtype=np.bool)
380384

385+
self.importance_history_ = imp_history
386+
381387
# notify user
382388
if self.verbose > 0:
383389
self._print_results(dec_reg, _iter, 1)

0 commit comments

Comments
 (0)