| 1331 | } |
| 1332 | |
| 1333 | int LGBM_BoosterCalcNumPredict(BoosterHandle handle, |
| 1334 | int num_row, |
| 1335 | int predict_type, |
| 1336 | int num_iteration, |
| 1337 | int64_t* out_len) { |
| 1338 | API_BEGIN(); |
| 1339 | Booster* ref_booster = reinterpret_cast<Booster*>(handle); |
| 1340 | *out_len = static_cast<int64_t>(num_row) * ref_booster->GetBoosting()->NumPredictOneRow( |
| 1341 | num_iteration, predict_type == C_API_PREDICT_LEAF_INDEX, predict_type == C_API_PREDICT_CONTRIB); |
| 1342 | API_END(); |
| 1343 | } |
| 1344 | |
| 1345 | int LGBM_BoosterSetNumLabels(BoosterHandle handle, int num_labels) { |
| 1346 | API_BEGIN(); |
no test coverage detected