8 references to GetCategoricalSplitFeaturesAt
Microsoft.ML.FastTree (4)
RegressionTree.cs (2)
58
/// returned value with nodeIndex=i is NOT a sub-set of <see cref="
GetCategoricalSplitFeaturesAt
(int)"/> with
102
/// matche any of values returned by <see cref="
GetCategoricalSplitFeaturesAt
(int)"/>, we call it a
Utils\RegressionTreeBaseUtils.cs (2)
84
var categoricalSplitFeaturesArray = trees[i].
GetCategoricalSplitFeaturesAt
(j).ToArray();
86
var len = trees[i].
GetCategoricalSplitFeaturesAt
(j).ToArray().Length;
Microsoft.ML.IntegrationTests (2)
IntrospectiveTraining.cs (2)
62
Assert.Empty(tree.
GetCategoricalSplitFeaturesAt
(0));
106
Assert.Empty(tree.
GetCategoricalSplitFeaturesAt
(0));
Microsoft.ML.Tests (2)
TrainerEstimators\TreeEstimators.cs (2)
829
if (tree.
GetCategoricalSplitFeaturesAt
(nodeId).Count() > 0)
830
Assert.Equal(row.CategoricalSplitFeatures, tree.
GetCategoricalSplitFeaturesAt
(nodeId).ToArray());