Fix style check warnings

This commit is contained in:
Rishabh Garg
2021-07-12 10:36:33 +05:30
parent f0f78cc65a
commit 1a171bcfef
3 changed files with 8 additions and 4 deletions
@@ -409,7 +409,10 @@ class DecisionTreeRegressor :
size_t NumLeaves() const;
//! Get the child of the given index.
const DecisionTreeRegressor& Child(const size_t i) const { return *children[i]; }
const DecisionTreeRegressor& Child(const size_t i) const
{
return *children[i];
}
//! Modify the child of the given index (be careful!).
DecisionTreeRegressor& Child(const size_t i) { return *children[i]; }
@@ -677,7 +677,8 @@ double DecisionTreeRegressor<FitnessFunction,
// Get the number of children we will have.
size_t numChildren = 0;
if (datasetInfo.Type(bestDim) == data::Datatype::categorical)
numChildren = CategoricalSplit::NumChildren(splitPointOrPrediction, *this);
numChildren = CategoricalSplit::NumChildren(splitPointOrPrediction,
*this);
else
numChildren = NumericSplit::NumChildren(splitPointOrPrediction, *this);
@@ -837,7 +838,8 @@ double DecisionTreeRegressor<FitnessFunction,
if (bestDim != data.n_rows)
{
// We know that the split is numeric.
size_t numChildren = NumericSplit::NumChildren(splitPointOrPrediction, *this);
size_t numChildren = NumericSplit::NumChildren(splitPointOrPrediction,
*this);
splitDimension = bestDim;
dimensionType = (size_t) data::Datatype::numeric;
@@ -98,7 +98,6 @@ class MADGain
return Evaluate<UseWeights>(values, weights, 0, values.n_elem);
}
};
} // namespace tree