Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix calling GPU predictor (#4836)

Browse Source 
* Fix calling GPU predictor
This commit is contained in:
Jiaming Yuan
2019-09-05 19:09:38 -04:00
committed by GitHub
parent 52d44e07fe
commit a5f232feb8
5 changed files with 85 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/predictor/cpu_predictor.cc
View File

@@ -134,7 +134,7 @@ class CPUPredictor : public Predictor {
} else {
if (!base_margin.empty()) {
std::ostringstream oss;
oss << "Warning: Ignoring the base margin, since it has incorrect length. "
oss << "Ignoring the base margin, since it has incorrect length. "
<< "The base margin must be an array of length ";
if (model.param.num_output_group > 1) {
oss << "[num_class] * [number of data points], i.e. "
@@ -145,7 +145,7 @@ class CPUPredictor : public Predictor {
}
oss << "Instead, all data points will use "
<< "base_score = " << model.base_margin;
LOG(INFO) << oss.str();
LOG(WARNING) << oss.str();
}
std::fill(out_preds_h.begin(), out_preds_h.end(), model.base_margin);
}