6 instantiations of TrainContext
Microsoft.ML.Core (2)
Prediction\ITrainer.cs (2)
94=> trainer.Train(new TrainContext(trainData)); 105=> trainer.Train(new TrainContext(trainData));
Microsoft.ML.Data (2)
Commands\TrainCommand.cs (1)
280var predictor = trainer.Train(new TrainContext(data, validData, testData, inputPredictor));
Training\TrainerEstimatorBase.cs (1)
157var pred = TrainModelCore(new TrainContext(trainRoleMapped, validRoleMapped, null, initPredictor));
Microsoft.ML.StandardTrainers (1)
Standard\Simple\SimpleTrainers.cs (1)
246var pred = ((ITrainer<PriorModelParameters>)this).Train(new TrainContext(trainRoles));
Microsoft.ML.Tests (1)
Scenarios\Api\Estimators\TrainWithInitialPredictor.cs (1)
44var finalModel = ((ITrainer)secondTrainer).Train(new TrainContext(trainRoles, initialPredictor: firstModel.Model));
53 references to TrainContext
Microsoft.ML.Core (12)
Prediction\ITrainer.cs (7)
61/// <seealso cref="ITrainer{TPredictor}.Train(TrainContext)"/> 62IPredictor Train(TrainContext context); 79new TPredictor Train(TrainContext context); 87/// Equivalent to calling <see cref="ITrainer.Train(TrainContext)"/> 88/// on a <see cref="TrainContext"/> constructed with <paramref name="trainData"/>. 98/// Equivalent to calling <see cref="ITrainer{TPredictor}.Train(TrainContext)"/> 99/// on a <see cref="TrainContext"/> constructed with <paramref name="trainData"/>.
Prediction\TrainContext.cs (2)
12/// into <see cref="ITrainer{TPredictor}.Train(TrainContext)"/> or <see cref="ITrainer.Train(TrainContext)"/>.
Prediction\TrainerInfo.cs (3)
39/// Whether the trainer supports validation set via <see cref="TrainContext.ValidationSet"/>. Not implementing 47/// Whether the trainer can use test set via <see cref="TrainContext.TestSet"/>. Not implementing 55/// Whether the trainer can support incremental trainers via <see cref="TrainContext.InitialPredictor"/>. Not
Microsoft.ML.Data (6)
TrainCatalog.cs (1)
143/// Subclasses of <see cref="TrainContext"/> will provide little "extension method" hookable objects
Training\TrainerEstimatorBase.cs (3)
97IPredictor ITrainer<IPredictor>.Train(TrainContext context) 161private protected abstract TModel TrainModelCore(TrainContext trainContext); 168IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);
Training\TrainerUtils.cs (2)
422public TPredictor Train(TrainContext context) 455IPredictor ITrainer.Train(TrainContext context) => Train(context);
Microsoft.ML.Ensemble (2)
Trainer\EnsembleTrainerBase.cs (2)
108IPredictor ITrainer<IPredictor>.Train(TrainContext context) 118IPredictor ITrainer.Train(TrainContext context)
Microsoft.ML.FastTree (9)
FastTreeClassification.cs (1)
184private protected override CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> TrainModelCore(TrainContext context)
FastTreeRanking.cs (1)
138private protected override FastTreeRankingModelParameters TrainModelCore(TrainContext context)
FastTreeRegression.cs (1)
108private protected override FastTreeRegressionModelParameters TrainModelCore(TrainContext context)
FastTreeTweedie.cs (1)
117private protected override FastTreeTweedieModelParameters TrainModelCore(TrainContext context)
GamClassification.cs (1)
137private protected override CalibratedModelParametersBase<GamBinaryModelParameters, PlattCalibrator> TrainModelCore(TrainContext context)
GamRegression.cs (1)
106private protected override GamRegressionModelParameters TrainModelCore(TrainContext context)
GamTrainer.cs (1)
220private protected void TrainBase(TrainContext context)
RandomForestClassification.cs (1)
211private protected override FastForestBinaryModelParameters TrainModelCore(TrainContext context)
RandomForestRegression.cs (1)
350private protected override FastForestRegressionModelParameters TrainModelCore(TrainContext context)
Microsoft.ML.KMeansClustering (1)
KMeansPlusPlusTrainer.cs (1)
197private protected override KMeansModelParameters TrainModelCore(TrainContext context)
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
384private protected override TModel TrainModelCore(TrainContext context)
Microsoft.ML.Mkl.Components (2)
OlsLinearRegression.cs (1)
151private protected override OlsModelParameters TrainModelCore(TrainContext context)
SymSgdClassificationTrainer.cs (1)
217private protected override TPredictor TrainModelCore(TrainContext context)
Microsoft.ML.PCA (1)
PcaTrainer.cs (1)
186private protected override PcaModelParameters TrainModelCore(TrainContext context)
Microsoft.ML.Recommender (4)
MatrixFactorizationTrainer.cs (4)
422/// <param name="context">The information collection needed for training. <see cref="TrainContext"/> for details.</param> 423private MatrixFactorizationModelParameters Train(TrainContext context) 432IPredictor ITrainer.Train(TrainContext context) => Train(context); 433MatrixFactorizationModelParameters ITrainer<MatrixFactorizationModelParameters>.Train(TrainContext context) => Train(context);
Microsoft.ML.StandardTrainers (14)
FactorizationMachine\FactorizationMachineTrainer.cs (3)
553private FieldAwareFactorizationMachineModelParameters Train(TrainContext context) 567IPredictor ITrainer.Train(TrainContext context) => Train(context); 568FieldAwareFactorizationMachineModelParameters ITrainer<FieldAwareFactorizationMachineModelParameters>.Train(TrainContext context) => Train(context);
LdSvm\LdSvmTrainer.cs (1)
189private protected override LdSvmModelParameters TrainModelCore(TrainContext trainContext)
Standard\LogisticRegression\LbfgsPredictorBase.cs (1)
430private protected override TModel TrainModelCore(TrainContext context)
Standard\MulticlassClassification\MetaMulticlassTrainer.cs (2)
117IPredictor ITrainer<IPredictor>.Train(TrainContext context) 184IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);
Standard\MulticlassClassification\MulticlassNaiveBayesTrainer.cs (1)
132private protected override NaiveBayesMulticlassModelParameters TrainModelCore(TrainContext context)
Standard\Online\OnlineLinear.cs (1)
274private protected sealed override TModel TrainModelCore(TrainContext context)
Standard\SdcaBinary.cs (1)
69private protected override TModel TrainModelCore(TrainContext context)
Standard\Simple\SimpleTrainers.cs (3)
250private PriorModelParameters Train(TrainContext context) 296IPredictor ITrainer.Train(TrainContext context) => Train(context); 298PriorModelParameters ITrainer<PriorModelParameters>.Train(TrainContext context) => Train(context);
Standard\StochasticTrainerBase.cs (1)
30private protected override TModel TrainModelCore(TrainContext context)
Microsoft.ML.Vision (1)
ImageClassificationTrainer.cs (1)
654private protected override ImageClassificationModelParameters TrainModelCore(TrainContext trainContext)