! * \brief Shrinkage for the tree's output * shrinkage rate (a.k.a learning rate) is used to tune the training process * \param rate The factor of shrinkage */
| 147 | * \param rate The factor of shrinkage |
| 148 | */ |
| 149 | inline void Shrinkage(double rate) { |
| 150 | #pragma omp parallel for schedule(static, 1024) if (num_leaves_ >= 2048) |
| 151 | for (int i = 0; i < num_leaves_; ++i) { |
| 152 | leaf_value_[i] *= rate; |
| 153 | } |
| 154 | shrinkage_ *= rate; |
| 155 | } |
| 156 | |
| 157 | inline double shrinkage() const { |
| 158 | return shrinkage_; |
no outgoing calls
no test coverage detected