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

Add additional Python tests to test training under constraints (#4426)

Browse Source
This commit is contained in:
Philip Hyunsu Cho
2019-04-30 18:23:39 -07:00
committed by GitHub
parent eaab364a63
commit ba98e0cdf2
3 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/tree/split_evaluator.cc
View File

@@ -499,8 +499,12 @@ class InteractionConstraint final : public SplitEvaluator {
// permissible in a given node; returns false otherwise
inline bool CheckInteractionConstraint(bst_uint featureid, bst_uint nodeid) const {
// short-circuit if no constraint is specified
return (params_.interaction_constraints.empty()
|| int_cont_[nodeid].count(featureid) > 0);
if (params_.interaction_constraints.empty()) {
return true;
}
CHECK_LT(nodeid, int_cont_.size()) << "Invariant violated: nodeid = "
<< nodeid << ", int_cont_.size() = " << int_cont_.size();
return (int_cont_[nodeid].count(featureid) > 0);
}
};