7 instantiations of InternalTreeEnsemble
Microsoft.ML.FastTree (7)
FastTree.cs (2)
470Ensemble = new InternalTreeEnsemble(); 2874TrainedEnsemble = new InternalTreeEnsemble(ctx, usingDefaultValues, categoricalSplits);
GamModelParameters.cs (1)
439var ensemble = new InternalTreeEnsemble();
RandomForestClassification.cs (1)
311TrainedEnsemble = new InternalTreeEnsemble();
RandomForestRegression.cs (1)
443TrainedEnsemble = new InternalTreeEnsemble();
Training\EnsembleCompression\LassoBasedEnsembleCompressor.cs (1)
463InternalTreeEnsemble ensemble = new InternalTreeEnsemble();
TreeEnsemble\TreeEnsembleCombiner.cs (1)
41var ensemble = new InternalTreeEnsemble();
36 references to InternalTreeEnsemble
Microsoft.ML.FastTree (35)
FastTree.cs (9)
55private protected InternalTreeEnsemble TrainedEnsemble; 86private protected InternalTreeEnsemble Ensemble; 915/// <seealso cref="InternalTreeEnsemble.RemapFeatures"/> 2789internal InternalTreeEnsemble TrainedEnsemble { get; } 2836private protected TreeEnsembleModelParameters(IHostEnvironment env, string name, InternalTreeEnsemble trainedEnsemble, int numFeatures, string innerArgs) 3366/// <see cref="InternalTreeEnsemble"/> in <see cref="ML.Trainers.FastTree.TreeEnsemble{T}"/>. 3371private protected TreeEnsembleModelParametersBasedOnRegressionTree(IHostEnvironment env, string name, InternalTreeEnsemble trainedEnsemble, int numFeatures, string innerArgs) 3412/// <see cref="InternalTreeEnsemble"/> in <see cref="ML.Trainers.FastTree.TreeEnsemble{T}"/>. 3417private protected TreeEnsembleModelParametersBasedOnQuantileRegressionTree(IHostEnvironment env, string name, InternalTreeEnsemble trainedEnsemble, int numFeatures, string innerArgs)
FastTreeClassification.cs (1)
71internal FastTreeBinaryModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs)
FastTreeRanking.cs (1)
1053internal FastTreeRankingModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs)
FastTreeRegression.cs (1)
501internal FastTreeRegressionModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs)
FastTreeTweedie.cs (1)
511internal FastTreeTweedieModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs)
GamModelParameters.cs (1)
439var ensemble = new InternalTreeEnsemble();
RandomForestClassification.cs (1)
91internal FastForestBinaryModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs)
RandomForestRegression.cs (1)
177internal FastForestRegressionModelParameters(IHostEnvironment env, InternalTreeEnsemble trainedEnsemble, int featureCount, string innerArgs, int samplesCount)
Training\BaggingProvider.cs (1)
78internal void ScaleEnsembleLeaves(int numTrees, int bagSize, InternalTreeEnsemble ensemble)
Training\EnsembleCompression\IEnsembleCompressor.cs (2)
14bool Compress(IChannel ch, InternalTreeEnsemble ensemble, double[] trainScores, int bestIteration, int maxTreesAfterCompression); 16InternalTreeEnsemble GetCompressedEnsemble();
Training\EnsembleCompression\LassoBasedEnsembleCompressor.cs (6)
54private InternalTreeEnsemble _compressedEnsemble; 461private InternalTreeEnsemble GetEnsembleFromSolution(LassoFit fit, int solutionIdx, InternalTreeEnsemble originalEnsemble) 463InternalTreeEnsemble ensemble = new InternalTreeEnsemble(); 537bool IEnsembleCompressor<short>.Compress(IChannel ch, InternalTreeEnsemble ensemble, double[] trainScores, int bestIteration, int maxTreesAfterCompression) 555InternalTreeEnsemble IEnsembleCompressor<short>.GetCompressedEnsemble()
Training\OptimizationAlgorithms\AcceleratedGradientDescent.cs (1)
11internal AcceleratedGradientDescent(InternalTreeEnsemble ensemble, Dataset trainData, double[] initTrainScores, IGradientAdjuster gradientWrapper)
Training\OptimizationAlgorithms\ConjugateGradientDescent.cs (1)
15public ConjugateGradientDescent(InternalTreeEnsemble ensemble, Dataset trainData, double[] initTrainScores, IGradientAdjuster gradientWrapper)
Training\OptimizationAlgorithms\GradientDescent.cs (1)
25internal GradientDescent(InternalTreeEnsemble ensemble, Dataset trainData, double[] initTrainScores, IGradientAdjuster gradientWrapper)
Training\OptimizationAlgorithms\NoOptimizationAlgorithm.cs (1)
16internal RandomForestOptimizer(InternalTreeEnsemble ensemble, Dataset trainData, double[] initTrainScores, IGradientAdjuster gradientWrapper)
Training\OptimizationAlgorithms\OptimizationAlgorithm.cs (2)
29public InternalTreeEnsemble Ensemble; 40public OptimizationAlgorithm(InternalTreeEnsemble ensemble, Dataset trainData, double[] initTrainScores)
TreeEnsemble\InternalRegressionTree.cs (1)
623/// Weight of this tree in an <see cref="InternalTreeEnsemble"/>.
TreeEnsemble\InternalTreeEnsemble.cs (1)
23/// String appended to the text representation of <see cref="InternalTreeEnsemble"/>. This is mainly used in <see cref="ToTreeEnsembleIni"/>.
TreeEnsemble\TreeEnsembleCombiner.cs (1)
41var ensemble = new InternalTreeEnsemble();
Utils\FastTreeIniFileUtils.cs (1)
16IHost host, InternalTreeEnsemble ensemble, RoleMappedSchema schema, ICalibrator calibrator,
Microsoft.ML.Sweeper (1)
Algorithms\SmacSweeper.cs (1)
337var e = forest.TrainedEnsemble;