1 interface inheriting from ITrainer
Microsoft.ML.Core (1)
Prediction\ITrainer.cs (1)
71internal interface ITrainer<out TPredictor> : ITrainer
66 references to ITrainer
Microsoft.ML.Core (5)
Prediction\IPredictor.cs (1)
12/// <see cref="ITrainer"/> and <see cref="IPredictor"/> it is still useful, but for things based on
Prediction\ITrainer.cs (2)
87/// Equivalent to calling <see cref="ITrainer.Train(TrainContext)"/> 93public static IPredictor Train(this ITrainer trainer, RoleMappedData trainData)
Prediction\TrainContext.cs (2)
12/// into <see cref="ITrainer{TPredictor}.Train(TrainContext)"/> or <see cref="ITrainer.Train(TrainContext)"/>. 39/// The initial predictor, for incremental training. Note that if a <see cref="ITrainer"/> implementor
Microsoft.ML.Data (36)
Commands\CrossValidationCommand.cs (6)
31public IComponentFactory<ITrainer> Trainer; 265private RoleMappedData CreateRoleMappedData(IHostEnvironment env, IChannel ch, IDataView data, ITrainer trainer) 351private readonly IComponentFactory<ITrainer> _trainer; 363private readonly Func<IHostEnvironment, IChannel, IDataView, ITrainer, RoleMappedData> _createExamples; 390Func<IHostEnvironment, IChannel, IDataView, ITrainer, RoleMappedData> createExamples, 483ITrainer trainer = _trainer.CreateComponent(host);
Commands\TrainCommand.cs (9)
66public IComponentFactory<ITrainer> Trainer; 89private readonly IComponentFactory<ITrainer> _trainer; 142ITrainer trainer = _trainer.CreateComponent(Host); 211public static void CheckTrainer(IExceptionContext ectx, IComponentFactory<ITrainer> trainer, string dataFile) 246public static IPredictor Train(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, 252public static IPredictor Train(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, RoleMappedData validData, 258private static IPredictor TrainCore(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, RoleMappedData validData, 442public static bool AddNormalizerIfNeeded(IHostEnvironment env, IChannel ch, ITrainer trainer, ref IDataView view, string featureColumn, NormalizeOption autoNorm) 491private static bool AddCacheIfWanted(IHostEnvironment env, IChannel ch, ITrainer trainer, ref RoleMappedData data, bool? cacheData)
Commands\TrainTestCommand.cs (2)
29public IComponentFactory<ITrainer> Trainer; 125ITrainer trainer = ImplOptions.Trainer.CreateComponent(Host);
EntryPoints\InputBase.cs (2)
59Func<ITrainer> createTrainer, 80var trainer = createTrainer();
Prediction\Calibrator.cs (2)
845ITrainer trainer, IPredictor predictor, RoleMappedSchema schema) 908int maxRows, ITrainer trainer, IPredictor predictor, RoleMappedData data)
Training\TrainerEstimatorBase.cs (2)
52PredictionKind ITrainer.PredictionKind => PredictionKind; 168IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);
Training\TrainerUtils.cs (5)
413_env.Assert(trainer is ITrainer); 415var oldTrainer = (ITrainer)trainer; 455IPredictor ITrainer.Train(TrainContext context) => Train(context); 471where T : ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel>, ITrainer
Transforms\TrainAndScoreTransformer.cs (4)
150public IComponentFactory<ITrainer> Trainer; 173/// <param name="trainer">The <see cref="ITrainer"/> object i.e. the learning algorithm that will be used for training the model.</param> 179ITrainer trainer, 226private static IDataTransform Create(IHostEnvironment env, Arguments args, ITrainer trainer, IDataView input, IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory)
Utilities\ComponentCreation.cs (4)
215public static ITrainer CreateTrainer<TArgs>(this IHostEnvironment env, TArgs arguments, out string loadName) 219return CreateCore<ITrainer, TArgs, SignatureTrainer>(env, arguments, out loadName); 222public static ITrainer CreateTrainer(this IHostEnvironment env, string settings, out string loadName) 225return CreateCore<ITrainer>(env, typeof(SignatureTrainer), settings, out loadName);
Microsoft.ML.Ensemble (2)
Trainer\EnsembleTrainerBase.cs (2)
72PredictionKind ITrainer.PredictionKind => PredictionKind; 118IPredictor ITrainer.Train(TrainContext context)
Microsoft.ML.FastTree (10)
FastTreeArguments.cs (9)
21internal interface IFastTreeTrainerFactory : IComponentFactory<ITrainer> 107ITrainer IComponentFactory<ITrainer>.CreateComponent(IHostEnvironment env) => new FastTreeBinaryTrainer(env, this); 151ITrainer IComponentFactory<ITrainer>.CreateComponent(IHostEnvironment env) => new FastTreeRegressionTrainer(env, this); 210ITrainer IComponentFactory<ITrainer>.CreateComponent(IHostEnvironment env) => new FastTreeTweedieTrainer(env, this); 308ITrainer IComponentFactory<ITrainer>.CreateComponent(IHostEnvironment env) => new FastTreeRankingTrainer(env, this);
TreeEnsembleFeaturizer.cs (1)
540public IComponentFactory<ITrainer> Trainer;
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
728ch.Assert(((ITrainer)this).PredictionKind != PredictionKind.MulticlassClassification || GbmOptions.ContainsKey("num_class"),
Microsoft.ML.Recommender (2)
MatrixFactorizationTrainer.cs (2)
304PredictionKind ITrainer.PredictionKind => PredictionKind.Recommendation; 432IPredictor ITrainer.Train(TrainContext context) => Train(context);
Microsoft.ML.ResultProcessor (1)
ResultProcessor.cs (1)
452IComponentFactory<ITrainer> trainer;
Microsoft.ML.StandardTrainers (7)
FactorizationMachine\FactorizationMachineTrainer.cs (3)
180PredictionKind ITrainer.PredictionKind => PredictionKind.BinaryClassification; 202TrainerInfo ITrainer.Info => _info; 567IPredictor ITrainer.Train(TrainContext context) => Train(context);
Standard\MulticlassClassification\MetaMulticlassTrainer.cs (2)
47PredictionKind ITrainer.PredictionKind => PredictionKind; 184IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);
Standard\Simple\SimpleTrainers.cs (2)
209PredictionKind ITrainer.PredictionKind => PredictionKind.BinaryClassification; 296IPredictor ITrainer.Train(TrainContext context) => Train(context);
Microsoft.ML.Tests (1)
Scenarios\Api\Estimators\TrainWithInitialPredictor.cs (1)
44var finalModel = ((ITrainer)secondTrainer).Train(new TrainContext(trainRoles, initialPredictor: firstModel.Model));
Microsoft.ML.Transforms (1)
LearnerFeatureSelection.cs (1)
276ITrainer trainer = options.Filter.CreateComponent(host);