1 write to NodeIndex
Microsoft.ML.FastTree (1)
Training\RegressionTreeNodeDocuments.cs (1)
28
NodeIndex
= nodeIndex;
16 references to NodeIndex
Microsoft.ML.FastTree (16)
Training\RegressionTreeNodeDocuments.cs (16)
22
public bool IsLeaf =>
NodeIndex
< 0;
34
_documentCount = Tree.GetNodesLeaves(
NodeIndex
).Sum(leaf => Partitioning.NumDocsInLeaf(leaf));
42
Tree.UpdateOutputWithDelta(~
NodeIndex
, delta);
56
return Tree.GetNodesLeaves(
NodeIndex
).Select(l => new RegressionTreeNodeDocuments(Tree, Partitioning, ~l));
63
if (
NodeIndex
< 0) //It is a leaf
65
documentsInNode = new HashSet<int>(Partitioning.DocumentsInLeaf(~
NodeIndex
));
70
foreach (var leaf in Tree.GetNodesLeaves(
NodeIndex
))
103
LteNode = new RecursiveRegressionTree(Tree, Partitioning, Tree.GetLteChildForNode(
NodeIndex
));
104
GtNode = new RecursiveRegressionTree(Tree, Partitioning, Tree.GetGtChildForNode(
NodeIndex
));
116
Tree.SetOutput(~
NodeIndex
, myOutput);
131
if (
NodeIndex
< 0)
132
return Tree.GetOutput(~
NodeIndex
);
151
Tree.SetOutput(~
NodeIndex
, myOutput);
169
if (
NodeIndex
< 0)
171
nodeCount = documentCount[~
NodeIndex
];
172
return Tree.GetOutput(~
NodeIndex
);