Ekeany / Boruta-Shap

A Tree based feature selection tool which combines both the Boruta feature selection algorithm with shapley values.
MIT License
559 stars 86 forks source link

Fix tentative.to_list bug #117

Closed GZuin closed 9 months ago

GZuin commented 1 year ago

What does this PR do?

Currently, if tentative is set to True, it raises an error because self.tentative can be a list, and not an array. Thus, to_list() does not work. This changes from to_list to list(), thus keeping the same behavior but fixing issues when self.tentative is already a list.