Advertisement
Guest User

Untitled

a guest
Jan 19th, 2018
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.54 KB | None | 0 0
  1. from sklearn.feature_selection import SelectKBest,SelectPercentile
  2.  
  3.  
  4. features_list = ['poi','salary', 'deferral_payments', 'total_payments',
  5.        'exercised_stock_options', 'bonus', 'restricted_stock',
  6.        'restricted_stock_deferred', 'total_stock_value', 'expenses',
  7.        'loan_advances', 'director_fees', 'deferred_income',
  8.        'long_term_incentive']
  9.  
  10. kbest = SelectKBest(k= 4)
  11. kbest.fit(features_train, labels_train)
  12. kbest_financial_feats = zip(features_list[1:], kbest.scores_, kbest.pvalues_, kbest.get_support())
  13. print sorted(kbest_financial_feats, key= lambda kbest_financial_feats: kbest_financial_feats[1], reverse= True)
  14.  
  15.  
  16. # output
  17. [('loan_advances', 2.7194355225362137, 0.10230285546818838, True), ('director_fees', 0.44209441196557453, 0.5076592317140044, True), ('total_payments', 0.38222224033137525, 0.53783608133444827, True), ('restricted_stock_deferred', 0.31098646703321225, 0.57833429274585102, True), ('deferral_payments', 0.13733982626793317, 0.71173179458219971, False), ('exercised_stock_options', 0.12830042038876935, 0.72096339272865384, False), ('total_stock_value', 0.095489886234481833, 0.75795996978801217, False), ('deferred_income', 0.054988982219314003, 0.81508322871264716, False), ('bonus', 0.042201153548052976, 0.83765862883975917, False), ('expenses', 0.038486299696967666, 0.84487199943396152, False), ('restricted_stock', 0.019093982688226523, 0.89037805973110706, False), ('salary', 0.007628120832847236, 0.93057838078076993, False), ('long_term_incentive', 0.0058908601173862964, 0.93897578396702075, False)]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement