1 write to Ensemble
Microsoft.ML.FastTree (1)
FastTree.cs (1)
470Ensemble = new InternalTreeEnsemble();
29 references to Ensemble
Microsoft.ML.FastTree (29)
BoostingFastTree.cs (5)
84optimizationAlgorithm = new GradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 87optimizationAlgorithm = new AcceleratedGradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 90optimizationAlgorithm = new ConjugateGradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 149bestIteration = Ensemble.NumTrees; 156int bestIteration = Ensemble.NumTrees;
FastTree.cs (17)
233TrainedEnsemble = Ensemble; 242bestIteration = Ensemble.NumTrees; 245private protected virtual int GetBestIteration(IChannel ch) => Ensemble.NumTrees; 330if (Ensemble.NumTrees == 0) 614if (Ensemble.NumTrees < numTotalTrees && ShouldRandomStartOptimizer()) 648pch.SetHeader(new ProgressHeader("trees"), e => e.SetProgress(0, Ensemble.NumTrees, numTotalTrees)); 649while (Ensemble.NumTrees < numTotalTrees) 661if (FastTreeTrainerOptions.BaggingSize > 0 && Ensemble.NumTrees % FastTreeTrainerOptions.BaggingSize == 0) 684else if (FastTreeTrainerOptions.BaggingSize > 0 && Ensemble.Trees.Count() > 0) 686ch.Assert(Ensemble.Trees.Last() == tree); 687Ensemble.Trees.Last() 751Ensemble.NumTrees, bestIteration); 762Ensemble.PopulateRawThresholds(TrainSet); 791int iteration = Ensemble.NumTrees; 798if (FastTreeTrainerOptions.TestFrequency != int.MaxValue && (Ensemble.NumTrees % FastTreeTrainerOptions.TestFrequency == 0 || Ensemble.NumTrees == FastTreeTrainerOptions.NumberOfTrees)) 864Ensemble.GetOutputs(set, scores);
FastTreeRanking.cs (4)
354lineBuilder.AppendFormat("Eval:\tnet.{0:D8}.ini", Ensemble.NumTrees - 1); 399double[] trainOutputs = Ensemble.GetTreeAt(Ensemble.NumTrees - 1).GetOutputs(TrainSet); 400_ensembleCompressor.SetTreeScores(Ensemble.NumTrees - 1, trainOutputs);
FastTreeRegression.cs (1)
377lineBuilder.AppendFormat("Eval:\tnet.{0:D8}.ini", Ensemble.NumTrees - 1);
FastTreeTweedie.cs (1)
313lineBuilder.AppendFormat("Eval:\tnet.{0:D8}.ini", Ensemble.NumTrees - 1);
RandomForest.cs (1)
46var optimizationAlgorithm = new RandomForestOptimizer(Ensemble, TrainSet, InitTrainScores, gradientWrapper);