7 references to SplitGains
Microsoft.ML.FastTree (3)
RegressionTree.cs (1)
49
/// See <see cref="
SplitGains
"/>.
Utils\RegressionTreeBaseUtils.cs (2)
78
splitGains.AddRange(trees[i].
SplitGains
.AsEnumerable());
109
builder.AddColumn(nameof(RegressionTreeBase.
SplitGains
), NumberDataViewType.Double, splitGains.ToArray());
Microsoft.ML.IntegrationTests (3)
IntrospectiveTraining.cs (3)
59
Assert.Equal(tree.
SplitGains
.Count, tree.NumberOfNodes);
103
Assert.Equal(tree.
SplitGains
.Count, tree.NumberOfNodes);
122
Assert.Equal(expectedSplitGains[i], finalTree.
SplitGains
[i], 0.000001);
Microsoft.ML.Tests (1)
TrainerEstimators\TreeEstimators.cs (1)
828
Assert.Equal(row.SplitGains, tree.
SplitGains
[nodeId]);