1 write to BestSplitInfoPerLeaf
Microsoft.ML.FastTree (1)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (1)
136BestSplitInfoPerLeaf = new SplitInfo[numLeaves];
14 references to BestSplitInfoPerLeaf
Microsoft.ML.FastTree (14)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (14)
212BestSplitInfoPerLeaf[0] = newRootSplitInfo; 239SplitInfo rootSplitInfo = BestSplitInfoPerLeaf[0]; 265bestLeaf = BestSplitInfoPerLeaf.Select(info => info.Gain).ArgMax(tree.NumLeaves); 266SplitInfo bestLeafSplitInfo = BestSplitInfoPerLeaf[bestLeaf]; 286SplitInfo bestSplitInfo = BestSplitInfoPerLeaf[bestLeaf]; 376_parallelTraining.FindGlobalBestSplit(SmallerChildSplitCandidates, null, FindBestThresholdFromRawArray, BestSplitInfoPerLeaf); 382/// and fills in the corresponding elements of <see cref="BestSplitInfoPerLeaf"/>. 398BestSplitInfoPerLeaf[lteChild].Gain = double.NegativeInfinity; 399BestSplitInfoPerLeaf[gtChild].Gain = double.NegativeInfinity; 440_parallelTraining.FindGlobalBestSplit(SmallerChildSplitCandidates, LargerChildSplitCandidates, FindBestThresholdFromRawArray, BestSplitInfoPerLeaf); 494BestSplitInfoPerLeaf[leaf] = leafSplitCandidates.FeatureSplitInfo[bestFeature]; 495if (BestSplitInfoPerLeaf[leaf].CategoricalSplit) 496Array.Sort(BestSplitInfoPerLeaf[leaf].CategoricalFeatureIndices); 498BestSplitInfoPerLeaf[leaf].Feature = bestFeature;