Advertisement
Guest User

Untitled

a guest
Jul 25th, 2018
138
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.48 KB | None | 0 0
  1. params = {
  2.     'n_estimators' : 10000,
  3.     'objective' : 'binary',
  4.     'learning_rate' : 0.05,
  5.     'reg_alpha' : 0.1,
  6.     'reg_lambda' : 0.1,
  7.     'seed' : 50,
  8.     'metric' : 'auc',
  9.     'verbose' : 100
  10. }
  11.  
  12. for i in range(2, 12):
  13.     train_bin = lgb.Dataset('train_bin_{}.bin'.format(i))
  14.     valid_bin = lgb.Dataset('valid_bin_{}.bin'.format(i))
  15.    
  16.     model = lgb.train(params, train_bin, valid_sets=[valid_bin], early_stopping_rounds=100)
  17.     print(model.best_iteration_)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement