Skip to content

Instantly share code, notes, and snippets.

@MLWhiz

MLWhiz/rfe.py

Created July 27, 2019 16:57
Show Gist options
  • Save MLWhiz/a1bfa1adc7e72ac0070b031bdd2e1859 to your computer and use it in GitHub Desktop.
Save MLWhiz/a1bfa1adc7e72ac0070b031bdd2e1859 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rfe.py
from sklearn.feature_selection import RFE
from sklearn.linear_model import LogisticRegression
rfe_selector = RFE(estimator=LogisticRegression(), n_features_to_select=num_feats, step=10, verbose=5)
rfe_selector.fit(X_norm, y)
rfe_support = rfe_selector.get_support()
rfe_feature = X.loc[:,rfe_support].columns.tolist()
print(str(len(rfe_feature)), 'selected features')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment