Skip to content

Instantly share code, notes, and snippets.

@Ekeany

Ekeany/Example2.py

Created May 26, 2020 16:02
Show Gist options
  • Save Ekeany/a3b64c8f4f3d46de7d730c477ec1d013 to your computer and use it in GitHub Desktop.
Save Ekeany/a3b64c8f4f3d46de7d730c477ec1d013 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Example2.py
# no model selected default is Random Forest, if classification is True it is a Classification problem
Feature_Selector = BorutaShap(importance_measure='shap',
classification=False)
Feature_Selector.fit(X=X, y=y, n_trials=100, random_state=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment