1 write to CategoricalThresholds
Microsoft.ML.FastTree (1)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (1)
301
CategoricalThresholds
= new HashSet<int>();
4 references to CategoricalThresholds
Microsoft.ML.FastTree (4)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (4)
300
if (
CategoricalThresholds
== null)
303
CategoricalThresholds
.Clear();
311
CategoricalThresholds
.Add(localIndex);
314
Partitioning.Split(bestLeaf, (TrainData.Flocks[bestSplitInfo.Flock] as OneHotFeatureFlock)?.Bins,
CategoricalThresholds
, gtChild);