Advertisement
Guest User

Untitled

a guest
Aug 18th, 2019
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.55 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.model_selection import train_test_split
  3. from sklearn.feature_selection import RFE
  4. from sklearn.linear_model import LogisticRegression
  5.  
  6. full_df = pd.read_csv('data.csv')
  7.  
  8. x = full_df.iloc[:,:-1]
  9. y = full_df.iloc[:,-1]
  10.  
  11. x_train, x_test, y_train, y_test = train_test_split(x, y, test_size = 0.3, random_state = 42)
  12.  
  13. model = LogisticRegression(solver ='lbfgs')
  14.  
  15. for i in range(1,120):
  16.  
  17. rfe = RFE(model, i)
  18. fit = rfe.fit(x_train, y_train)
  19. acc = fit.score(x_test, y_test)
  20.  
  21. print(acc)
  22. print(fit.support_)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement