Advertisement
Guest User

Untitled

a guest
Jun 20th, 2019
53
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.54 KB | None | 0 0
  1. import numpy as np
  2. from sklearn.ensemble import RandomForestClassifier
  3. from sklearn.feature_selection import RFE
  4.  
  5. X = np.asarray(df[['age', 'job', 'marital', 'education', 'housing_loan', 'personal_loan',
  6. 'emp.var.rate', 'cons.price.idx', 'cons.conf.idx', 'euribor3m']])
  7. y = np.asarray(df['y'])
  8. rfc = RandomForestClassifier(n_estimators=40)
  9. rfe = RFE(rfc, 6)
  10. rfe_fit = rfe.fit(X, y)
  11.  
  12. print("Num Features: %s" % (rfe_fit.n_features_))
  13. print("Selected Features: %s" % (rfe_fit.support_))
  14. print("Feature Ranking: %s" % (rfe_fit.ranking_))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement