2 writes to Dataset
Microsoft.ML.FastTree (2)
Training\ScoreTracker.cs (2)
23Dataset = s.Dataset; 35Dataset = set;
22 references to Dataset
Microsoft.ML.FastTree (22)
FastTree.cs (2)
688.AddOutputsToScores(OptimizationAlgorithm.TrainingScores.Dataset, 851if (st.Dataset == set)
FastTreeTweedie.cs (1)
405var weights = trainingScores.Dataset.SampleWeights;
Training\OptimizationAlgorithms\GradientDescent.cs (2)
89Dataset.DatasetSkeleton dsSkeleton = TrainingScores.Dataset.Skeleton; 137_treeScores.Add(tree.GetOutputs(TrainingScores.Dataset));
Training\OptimizationAlgorithms\OptimizationAlgorithm.cs (1)
89if (st.Dataset == set)
Training\ScoreTracker.cs (9)
23Dataset = s.Dataset; 62Scores = new double[Dataset.NumDocs]; 68if (initScores.Length != Dataset.NumDocs) 97tree.AddOutputsToScores(Dataset, Scores, multiplier); 172int innerLoopSize = 1 + Dataset.NumDocs / BlockingThreadPool.NumThreads; // +1 is to make sure we don't have a few left over at the end 175var actions = new Action[(int)Math.Ceiling(1.0 * Dataset.NumDocs / innerLoopSize)]; 177for (int d = 0; d < Dataset.NumDocs; d += innerLoopSize) 180var toDoc = Math.Min(d + innerLoopSize, Dataset.NumDocs); 183var featureBins = Dataset.GetFeatureBinRowwiseIndexer();
Training\StepSearch.cs (2)
83if (Scores == null || Scores.Dataset != _previousScores.Dataset)
Training\Test.cs (5)
147public Dataset Dataset => ScoreTracker.Dataset; 348Contracts.Check(scoreTracker.Dataset.NumDocs == labels.Length, "Mismatch between dataset and labels"); 527_labels = FastTreeRegressionTrainer.GetDatasetRegressionLabels(scoreTracker.Dataset); 528Contracts.Check(scoreTracker.Dataset.NumDocs == _labels.Length, "Mismatch between dataset and labels"); 602Contracts.Check(scoreTracker.Dataset.NumDocs == binaryLabels.Length, "Mismatch between dataset and labels");