Learn Before
Concept

Decision tree key parameters

  • N_estimators: sets # of small decision trees to use in the ensemble
    • More trees can lead to overfitting
    • Chosen to best exploit the speed and memory capabilities
  • Learning_rate: controls emphasis on fixing errors from previous iteration
    • Adjusted given n_estimators
  • The above two are typically tuned together
    • Small learning rate requires more n estimators to maintain model complexity
  • Max_depth: typically set to a small value (3-5) for most applications Can effect model complexity

0

1

Updated 2021-02-19

Tags

Data Science