2 implementations of GetLeafSplitGain
Microsoft.ML.FastTree (2)
GamTrainer.cs (1)
627
public double
GetLeafSplitGain
(int count, double sumTargets, double sumWeights)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (1)
608
public double
GetLeafSplitGain
(int count, double sumTargets, double sumWeights)
3 references to GetLeafSplitGain
Microsoft.ML.FastTree (3)
Dataset\FeatureFlock.cs (3)
252
double gainShift = leafCalculator.
GetLeafSplitGain
(totalCount, sumTargets, sumWeights);
288
double currentShiftedGain = leafCalculator.
GetLeafSplitGain
(gtCount, sumGTTargets, sumGTWeights)
289
+ leafCalculator.
GetLeafSplitGain
(lteCount, sumTargets - sumGTTargets, sumWeights - sumGTWeights);