! \brief Get training scores result */
| 695 | |
| 696 | /*! \brief Get training scores result */ |
| 697 | const double* GBDT::GetTrainingScore(int64_t* out_len) { |
| 698 | *out_len = static_cast<int64_t>(train_score_updater_->num_data()) * num_class_ * num_labels_; |
| 699 | return train_score_updater_->score(); |
| 700 | } |
| 701 | |
| 702 | void GBDT::PredictContrib(const double* features, double* output, const PredictionEarlyStopInstance* early_stop) const { |
| 703 | int early_stop_round_counter = 0; |