1 type derived from InternalRegressionTree
Microsoft.ML.FastTree (1)
TreeEnsemble\InternalQuantileRegressionTree.cs (1)
11internal class InternalQuantileRegressionTree : InternalRegressionTree
5 instantiations of InternalRegressionTree
Microsoft.ML.FastTree (5)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (1)
198return new InternalRegressionTree(NumLeaves);
TreeEnsemble\InternalRegressionTree.cs (3)
172InternalRegressionTree tree = new InternalRegressionTree(2); 188return new InternalRegressionTree(splitFeatures, splitGain, null, rawThresholds, defaultValueForMissing, lteChild, gtChild, leafValues, categoricalSplitFeatures, categoricalSplit); 424return new InternalRegressionTree(ctx, usingDefaultValues, categoricalSplits);
TreeEnsemble\TreeEnsembleCombiner.cs (1)
71var tNew = new InternalRegressionTree(bytes, ref position);
80 references to InternalRegressionTree
Microsoft.ML.FastTree (80)
FastTree.cs (8)
678var tree = OptimizationAlgorithm.TrainingIteration(ch, activeFeatures); 784private protected virtual void CustomizedTrainingIteration(InternalRegressionTree tree) 3065foreach (var tree in TrainedEnsemble.Trees) 3186foreach (InternalRegressionTree tree in TrainedEnsemble.Trees) 3202private void SaveTreeAsCode(InternalRegressionTree tree, TextWriter writer, in VBuffer<ReadOnlyMemory<char>> names) 3208private void ToCSharp(InternalRegressionTree tree, TextWriter writer, int node, in VBuffer<ReadOnlyMemory<char>> names) 3279private readonly InternalRegressionTree _regTree; 3281public Tree(InternalRegressionTree regTree)
FastTreeClassification.cs (1)
398public void AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree,
FastTreeRanking.cs (2)
394private protected override void CustomizedTrainingIteration(InternalRegressionTree tree) 928void IStepSearch.AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree, DocumentPartitioning partitioning,
FastTreeRegression.cs (1)
447public void AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree, DocumentPartitioning partitioning, ScoreTracker trainingScores)
FastTreeTweedie.cs (1)
401public void AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree, DocumentPartitioning partitioning, ScoreTracker trainingScores)
GamModelParameters.cs (4)
452var tree = CreateRegressionTree(numLeaves, splitFeatures, treeThresholds, lteChild, gtChild, effects); 458var interceptTree = CreateRegressionTree( 527private static InternalRegressionTree CreateRegressionTree( 531return InternalRegressionTree.Create(
RegressionTree.cs (5)
12/// A container base class for exposing <see cref="InternalRegressionTree"/>'s and 22private readonly InternalRegressionTree _tree; 158internal RegressionTreeBase(InternalRegressionTree tree) 174/// A container class for exposing <see cref="InternalRegressionTree"/>'s attributes to users. 181internal RegressionTree(InternalRegressionTree tree) : base(tree) { }
Training\BaggingProvider.cs (1)
83InternalRegressionTree tree = ensemble.GetTreeAt(t);
Training\DocumentPartitioning.cs (1)
54internal DocumentPartitioning(InternalRegressionTree tree, Dataset dataset)
Training\EnsembleCompression\LassoBasedEnsembleCompressor.cs (1)
471InternalRegressionTree tree = originalEnsemble.GetTreeAt(fit.Indices[i]);
Training\OptimizationAlgorithms\AcceleratedGradientDescent.cs (3)
21internal override InternalRegressionTree TrainingIteration(IChannel ch, bool[] activeFeatures) 32InternalRegressionTree tree = base.TrainingIteration(ch, activeFeatures); 55internal override void UpdateScores(ScoreTracker t, InternalRegressionTree tree)
Training\OptimizationAlgorithms\GradientDescent.cs (2)
94internal override InternalRegressionTree TrainingIteration(IChannel ch, bool[] activeFeatures) 98InternalRegressionTree tree = TreeLearner.FitTargets(ch, activeFeatures, AdjustTargetsAndSetWeights(ch));
Training\OptimizationAlgorithms\NoOptimizationAlgorithm.cs (2)
28internal override InternalRegressionTree TrainingIteration(IChannel ch, bool[] activeFeatures) 35InternalRegressionTree tree = ((RandomForestLeastSquaresTreeLearner)TreeLearner).FitTargets(ch, activeFeatures, weightedTargets,
Training\OptimizationAlgorithms\OptimizationAlgorithm.cs (4)
56internal abstract InternalRegressionTree TrainingIteration(IChannel ch, bool[] activeFeatures); 58internal virtual void UpdateAllScores(IChannel ch, InternalRegressionTree tree) 69internal virtual void UpdateScores(ScoreTracker t, InternalRegressionTree tree) 104protected virtual void SmoothTree(InternalRegressionTree tree, double smoothing)
Training\Parallel\IParallelTraining.cs (1)
132double[] GlobalMean(Dataset dataset, InternalRegressionTree tree, DocumentPartitioning partitioning, double[] weights, bool filterZeroLambdas);
Training\Parallel\SingleTrainer.cs (1)
49double[] IParallelTraining.GlobalMean(Dataset dataset, InternalRegressionTree tree, DocumentPartitioning partitioning, double[] weights, bool filterZeroLambdas)
Training\RegressionTreeNodeDocuments.cs (3)
18internal readonly InternalRegressionTree Tree; 24internal RegressionTreeNodeDocuments(InternalRegressionTree tree, DocumentPartitioning partitioning, int nodeIndex) 96internal RecursiveRegressionTree(InternalRegressionTree t, DocumentPartitioning p, int n)
Training\ScoreTracker.cs (5)
41internal void Initialize(ScoreTracker scores1, InternalRegressionTree tree, DocumentPartitioning partitioning, double multiplier) 95internal virtual void AddScores(InternalRegressionTree tree, double multiplier) 103internal virtual void AddScores(InternalRegressionTree tree, DocumentPartitioning partitioning, double multiplier) 167internal override void AddScores(InternalRegressionTree tree, double multiplier) 198internal override void AddScores(InternalRegressionTree tree, DocumentPartitioning partitioning, double multiplier)
Training\StepSearch.cs (4)
13void AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree, DocumentPartitioning partitioning, ScoreTracker trainingScores); 62private InternalRegressionTree _tree; 66public void Initialize(InternalRegressionTree tree, DocumentPartitioning partitioning, ScoreTracker previousScores) 97void IStepSearch.AdjustTreeOutputs(IChannel ch, InternalRegressionTree tree, DocumentPartitioning partitioning,
Training\TreeLearners\FastForestLeastSquaresTreeLearner.cs (2)
27protected override InternalRegressionTree NewTree() 32public InternalRegressionTree FitTargets(IChannel ch, bool[] activeFeatures, Double[] weightedtargets, Double[] targets, Double[] weights)
Training\TreeLearners\LeastSquaresRegressionTreeLearner.cs (5)
196protected virtual InternalRegressionTree NewTree() 201protected virtual void MakeDummyRootSplit(InternalRegressionTree tree, double rootTarget, double[] targets) 222internal sealed override InternalRegressionTree FitTargets(IChannel ch, bool[] activeFeatures, double[] targets) 228InternalRegressionTree tree = NewTree(); 284protected virtual void PerformSplit(InternalRegressionTree tree, int bestLeaf, double[] targets, out int lteChild, out int gtChild)
Training\TreeLearners\TreeLearner.cs (1)
26internal abstract InternalRegressionTree FitTargets(IChannel ch, bool[] activeFeatures, double[] targets);
TreeEnsemble\InternalRegressionTree.cs (6)
20/// Note that <see cref="InternalRegressionTree"/> is shared between FastTree and LightGBM assemblies, 21/// so <see cref="InternalRegressionTree"/> has <see cref="BestFriendAttribute"/>. 165public static InternalRegressionTree Create(int numLeaves, int[] splitFeatures, double[] splitGain, 172InternalRegressionTree tree = new InternalRegressionTree(2); 418public static InternalRegressionTree Load(ModelLoadContext ctx, bool usingDefaultValues, bool categoricalSplits) 556public void SumupValue(IChannel ch, InternalRegressionTree tree)
TreeEnsemble\InternalTreeEnsemble.cs (13)
26private readonly List<InternalRegressionTree> _trees; 28public IEnumerable<InternalRegressionTree> Trees => _trees; 36_trees = new List<InternalRegressionTree>(); 50_trees = new List<InternalRegressionTree>(); 54AddTree(InternalRegressionTree.Load(ctx, usingDefaultValues, categoricalSplits)); 69foreach (InternalRegressionTree tree in Trees) 75public void AddTree(InternalRegressionTree tree) => _trees.Add(tree); 76public void AddTreeAt(InternalRegressionTree tree, int index) => _trees.Insert(index, tree); 80public InternalRegressionTree GetTreeAt(int index) => _trees[index]; 218foreach (var tree in Trees) 356foreach (var tree in _trees) 370var tree = _trees[i]; 399public FeatureToGainMap(IList<InternalRegressionTree> trees, bool normalize)
TreeEnsemble\TreeEnsembleCombiner.cs (3)
65foreach (var t in tree.TrainedEnsemble.Trees) 71var tNew = new InternalRegressionTree(bytes, ref position); 94foreach (var t in ensemble.Trees)