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

fix label crash

Browse Source
This commit is contained in:
tqchen 2015-04-06 08:48:06 -07:00
parent 529a732737
commit b6d85b9d9b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/learner/objective-inl.hpp
View File

@ -225,7 +225,8 @@ class SoftmaxMultiClassObj : public IObjFunction {
} }
} }
utils::Check(label_error >= 0 && label_error < nclass, utils::Check(label_error >= 0 && label_error < nclass,
"SoftmaxMultiClassObj: label must be in [0, num_class), found %d in label", label_error); "SoftmaxMultiClassObj: label must be in [0, num_class),"\
" num_class=%d but found %d in label", nclass, label_error);
} }
virtual void PredTransform(std::vector<float> *io_preds) { virtual void PredTransform(std::vector<float> *io_preds) {
this->Transform(io_preds, output_prob); this->Transform(io_preds, output_prob);