Advertisement
Guest User

Untitled

a guest
Aug 26th, 2019
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.62 KB | None | 0 0
  1. def cat_hyp(depth, bagging_temperature): # Function to optimize depth and bagging temperature
  2. params = {"iterations": 100,
  3. "learning_rate": 0.05,
  4. "eval_metric": "R2",
  5. "verbose": False} # Default Parameters
  6. params[ "depth"] = int(round(depth))
  7. params["bagging_temperature"] = bagging_temperature
  8.  
  9. cat_feat = [] # Categorical features list
  10. cv_dataset = cgb.Pool(data=X,
  11. label=y,
  12. cat_features=cat_feat)
  13.  
  14. scores = cgb.cv(cv_dataset,
  15. params,
  16. fold_count=3)
  17. return np.max(scores['test-R2-mean']) # Return maximum R-Squared value
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement