"vscode:/vscode.git/clone" did not exist on "d107872a5342442420edaa1d1849901f865f2331"
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