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

Add check for length of weights. (#4872)

Browse Source
This commit is contained in:
Jiaming Yuan
2019-12-21 11:30:58 +08:00
committed by GitHub
parent 3d04a8cc97
commit c8bdb652c4
4 changed files with 41 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/objective/hinge.cu
View File

@@ -37,8 +37,12 @@ class HingeObj : public ObjFunction {
<< "preds.size=" << preds.Size()
<< ", label.size=" << info.labels_.Size();
const bool is_null_weight = info.weights_.Size() == 0;
const size_t ndata = preds.Size();
const bool is_null_weight = info.weights_.Size() == 0;
if (!is_null_weight) {
CHECK_EQ(info.weights_.Size(), ndata)
<< "Number of weights should be equal to number of data points.";
}
out_gpair->Resize(ndata);
common::Transform<>::Init(
[=] XGBOOST_DEVICE(size_t _idx,