Unverified Commit 542c34dd authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix the negative dimension error in python predict

parent 941068ee
......@@ -224,7 +224,7 @@ public:
OMP_LOOP_EX_END();
}
OMP_THROW_EX();
*out_len = nrow * num_pred_in_one_row;
*out_len = num_pred_in_one_row * nrow;
}
void Predict(int num_iteration, int predict_type, const char* data_filename,
......@@ -1088,8 +1088,8 @@ int LGBM_BoosterCalcNumPredict(BoosterHandle handle,
int64_t* out_len) {
API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle);
*out_len = static_cast<int64_t>(num_row * ref_booster->GetBoosting()->NumPredictOneRow(
num_iteration, predict_type == C_API_PREDICT_LEAF_INDEX, predict_type == C_API_PREDICT_CONTRIB));
*out_len = static_cast<int64_t>(num_row) * ref_booster->GetBoosting()->NumPredictOneRow(
num_iteration, predict_type == C_API_PREDICT_LEAF_INDEX, predict_type == C_API_PREDICT_CONTRIB);
API_END();
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment