1 write to TrainSet
Microsoft.ML.FastTree (1)
FastTree.cs (1)
194TrainSet = instanceConverter.FindBinsAndReturnDataset(trainData, PredictionKind, ParallelTraining, CategoricalFeatures, FastTreeTrainerOptions.CategoricalSplit);
50 references to TrainSet
Microsoft.ML.FastTree (50)
BoostingFastTree.cs (4)
66TrainSet, FastTreeTrainerOptions.NumberOfLeaves, FastTreeTrainerOptions.MinimumExampleCountPerLeaf, FastTreeTrainerOptions.EntropyCoefficient, 84optimizationAlgorithm = new GradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 87optimizationAlgorithm = new AcceleratedGradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 90optimizationAlgorithm = new ConjugateGradientDescent(Ensemble, TrainSet, InitTrainScores, gradientWrapper);
FastTree.cs (10)
72/// In the training process, <see cref="TrainSet"/>, <see cref="ValidSet"/>, <see cref="TestSets"/> would be 412var activeFeatures = Utils.CreateArray(TrainSet.NumFeatures, true); 418for (int i = 0; i < TrainSet.NumFeatures; ++i) 439ch.Trace("Training {0}", GetDatasetStatistics(TrainSet)); 465return TrainSet.SampleWeights != null; 586return new BaggingProvider(TrainSet, FastTreeTrainerOptions.NumberOfLeaves, FastTreeTrainerOptions.Seed, FastTreeTrainerOptions.BaggingExampleFraction); 762Ensemble.PopulateRawThresholds(TrainSet); 827if (set == TrainSet) 828st = OptimizationAlgorithm.GetScoreTracker("train", TrainSet, InitTrainScores); 877if (set == TrainSet)
FastTreeClassification.cs (5)
219TrainSet, 253_trainSetLabels = GetClassificationLabelsFromRatings(TrainSet).ToArray(TrainSet.NumDocs); 259return new BinaryClassificationTest(ConstructScoreTracker(TrainSet), _trainSetLabels, _sigmoidParameter); 265TrainTest = new BinaryClassificationTest(ConstructScoreTracker(TrainSet), _trainSetLabels, _sigmoidParameter);
FastTreeRanking.cs (11)
201_ensembleCompressor.Initialize(FastTreeTrainerOptions.NumberOfTrees, TrainSet, TrainSet.Ratings, FastTreeTrainerOptions.Seed); 207return new LambdaRankObjectiveFunction(TrainSet, TrainSet.Ratings, FastTreeTrainerOptions, ParallelTraining); 215_specialTrainSetTest = new FastNdcgTest(optimizationAlgorithm.TrainingScores, TrainSet.Ratings, FastTreeTrainerOptions.SortingAlgorithm, FastTreeTrainerOptions.EarlyStoppingMetrics); 224return new RankingBaggingProvider(TrainSet, FastTreeTrainerOptions.NumberOfLeaves, FastTreeTrainerOptions.Seed, FastTreeTrainerOptions.BaggingExampleFraction); 233return new NdcgTest(ConstructScoreTracker(TrainSet), TrainSet.Ratings, FastTreeTrainerOptions.SortingAlgorithm); 282Tests.Add(CreateStandardTest(TrainSet)); 399double[] trainOutputs = Ensemble.GetTreeAt(Ensemble.NumTrees - 1).GetOutputs(TrainSet); 429TrainSet.Ratings,
FastTreeRegression.cs (5)
148return new ObjectiveImpl(TrainSet, FastTreeTrainerOptions); 188return new RegressionTest(ConstructScoreTracker(TrainSet)); 212Tests.Add(new RegressionTest(ConstructScoreTracker(TrainSet))); 276Test trainRegressionTest = new RegressionTest(ConstructScoreTracker(TrainSet)); 284TrainTest = new RegressionTest(ConstructScoreTracker(TrainSet), FastTreeTrainerOptions.EarlyStoppingMetrics);
FastTreeTweedie.cs (5)
160return new ObjectiveImpl(TrainSet, FastTreeTrainerOptions); 202return new RegressionTest(ConstructScoreTracker(TrainSet)); 218Tests.Add(new RegressionTest(ConstructScoreTracker(TrainSet))); 255Test trainRegressionTest = new RegressionTest(ConstructScoreTracker(TrainSet)); 263TrainTest = new RegressionTest(ConstructScoreTracker(TrainSet), FastTreeTrainerOptions.EarlyStoppingMetrics);
RandomForest.cs (2)
46var optimizationAlgorithm = new RandomForestOptimizer(Ensemble, TrainSet, InitTrainScores, gradientWrapper); 66TrainSet, FastTreeTrainerOptions.NumberOfLeaves, FastTreeTrainerOptions.MinimumExampleCountPerLeaf, FastTreeTrainerOptions.EntropyCoefficient,
RandomForestClassification.cs (5)
343newTree.PopulateThresholds(TrainSet); 350return new ObjectiveFunctionImpl(TrainSet, _trainSetLabels, FastTreeTrainerOptions); 357_trainSetLabels = TrainSet.Ratings.Select(x => x >= 1).ToArray(TrainSet.NumDocs); 362return new BinaryClassificationTest(ConstructScoreTracker(TrainSet), _trainSetLabels, 1);
RandomForestRegression.cs (3)
475newTree.PopulateThresholds(TrainSet); 486return ObjectiveFunctionImplBase.Create(TrainSet, FastTreeTrainerOptions); 491return new RegressionTest(ConstructScoreTracker(TrainSet));