11 references to CategoricalSplitFlags
Microsoft.ML.FastTree (8)
RegressionTree.cs (6)
41
/// See <see cref="
CategoricalSplitFlags
"/>.
60
/// Note that the case (1) happens only when <see cref="
CategoricalSplitFlags
"/>[i] is false and otherwise (2)
78
/// i-th node. This value is valid only if <see cref="
CategoricalSplitFlags
"/>[i] is false.
90
/// Determine the types of splitting function. If <see cref="
CategoricalSplitFlags
"/>[i] is true, the i-th
104
/// should go next. The returned value is valid only if <see cref="
CategoricalSplitFlags
"/>[nodeIndex] is true.
121
/// The returned value is valid only if <see cref="
CategoricalSplitFlags
"/>[nodeIndex] is true.
Utils\RegressionTreeBaseUtils.cs (2)
70
categoricalSplitFlags.AddRange(trees[i].
CategoricalSplitFlags
.AsEnumerable());
107
builder.AddColumn(nameof(RegressionTreeBase.
CategoricalSplitFlags
), BooleanDataViewType.Instance, categoricalSplitFlags.ToArray());
Microsoft.ML.IntegrationTests (2)
IntrospectiveTraining.cs (2)
61
Assert.All(tree.
CategoricalSplitFlags
, flag => Assert.False(flag));
105
Assert.All(tree.
CategoricalSplitFlags
, flag => Assert.False(flag));
Microsoft.ML.Tests (1)
TrainerEstimators\TreeEstimators.cs (1)
826
Assert.Equal(row.CategoricalSplitFlags, tree.
CategoricalSplitFlags
[nodeId]);