Advertisement
Guest User

Untitled

a guest
Mar 27th, 2017
46
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.83 KB | None | 0 0
  1. #LOADING THE DATASET
  2. X=sio.loadmat('/home/set.mat')['x']
  3. s_y=sio.loadmat('/home/set.mat')['y']
  4. y=np.ravel(s_y).astype(int)
  5.  
  6.  
  7. #PIPELINE
  8. clf = Pipeline([('rcl', RobustScaler()),
  9. ('clf', RandomForestClassifier(random_state=0, n_jobs=-1))])
  10.  
  11.  
  12. #OPTIMIZATION
  13. sss_outer = StratifiedShuffleSplit(n_splits=10, test_size=0.1, random_state=0)
  14. sss_inner = StratifiedShuffleSplit(n_splits=10, test_size=0.1, random_state=0)
  15.  
  16.  
  17. parameters = {'clf__n_estimators': sp_randint(170, 300),
  18. 'clf__max_features': sp_randint(5, 20),
  19. }
  20. n_iter_search = 30
  21.  
  22.  
  23. inner_rs = RandomizedSearchCV(clf, param_distributions=parameters, n_iter=n_iter_search, cv=sss_inner, random_state=0, n_jobs=-1)
  24. inner_rs.fit(X,y)
  25. outer_rs = cross_val_score(inner_rs, X, y, cv=sss_outer)
  26.  
  27. print(inner_rs.best_params_)
  28. print(outer_rs.mean())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement