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

Allow dataframes for Y #88

Closed sboshin closed 1 year ago

sboshin commented 2 years ago

Currently the error is incorrect, Y can only be a Pandas Series or an nd.array. Allow for dataframes, and series, and fix the Error message

What does this PR do?

Fixes a bug where Y can't be a dataframe, but the message says it can. Adds support for Y as a dataframe. References

Testing performed

Known issues