ITrainerEstimator

Training\TrainerEstimatorBase.cs (1)

16public abstract class TrainerEstimatorBase<TTransformer, TModel> : ITrainerEstimator<TTransformer, TModel>, ITrainer<IPredictor>

MatrixFactorizationTrainer.cs (1)

114ITrainerEstimator<MatrixFactorizationPredictionTransformer, MatrixFactorizationModelParameters>

Standard\MulticlassClassification\MetaMulticlassTrainer.cs (1)

17public abstract class MetaMulticlassTrainer<TTransformer, TModel> : ITrainerEstimator<TTransformer, TModel>, ITrainer<IPredictor>

Standard\Simple\SimpleTrainers.cs (1)

194ITrainerEstimator<BinaryPredictionTransformer<PriorModelParameters>, PriorModelParameters>

Experiment\SuggestedPipeline.cs (1)

133var learner = Trainer.BuildTrainer(validationSet);

Experiment\SuggestedTrainer.cs (1)

44public ITrainerEstimator<IPredictionTransformer<object>, object> BuildTrainer(IDataView validationSet = null)

TrainerExtensions\BinaryTrainerExtensions.cs (10)

TrainerExtensions\ITrainerExtension.cs (2)

10using ITrainerEstimator = ITrainerEstimator<IPredictionTransformer<object>, object>; 16ITrainerEstimator CreateInstance(MLContext mlContext,

TrainerExtensions\MultiTrainerExtensions.cs (12)

TrainerExtensions\RankingTrainerExtensions.cs (3)

13using ITrainerEstimator = ITrainerEstimator<IPredictionTransformer<object>, object>; 21public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable<SweepableParam> sweepParams, 44public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable<SweepableParam> sweepParams,

TrainerExtensions\RecommendationTrainerExtensions.cs (2)

10using ITrainerEsitmator = ITrainerEstimator<IPredictionTransformer<object>, object>; 14public ITrainerEsitmator CreateInstance(MLContext mlContext, IEnumerable<SweepableParam> sweepParams,

TrainerExtensions\RegressionTrainerExtensions.cs (9)

TrainerExtensionsTests.cs (2)

43var instance = extension.CreateInstance(context, sweepParams, columnInfo); 57var instance = extension.CreateInstance(context, null, columnInfo);

Training\TrainerUtils.cs (5)

392/// This is a shim class to translate the more contemporaneous <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> 405private readonly ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel> _trainer; 408public TrainerEstimatorToTrainerShim(IHostEnvironment env, ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel> trainer) 459/// This is a shim for legacy code that takes the more modern <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> 471where T : ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel>, ITrainer

OutputCombiners\BaseStacking.cs (3)

25internal abstract IComponentFactory<ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<TOutput>>, IPredictorProducing<TOutput>>> GetPredictorFactory(); 28private protected readonly IComponentFactory<ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<TOutput>>, IPredictorProducing<TOutput>>> BasePredictorType; 144var trainer = BasePredictorType.CreateComponent(host);

OutputCombiners\MultiStacking.cs (3)

22using TVectorTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<VBuffer<float>>>, IPredictorProducing<VBuffer<float>>>; 48public IComponentFactory<TVectorTrainer> BasePredictorType; 50internal override IComponentFactory<TVectorTrainer> GetPredictorFactory() => BasePredictorType;

OutputCombiners\RegressionStacking.cs (3)

20using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 46public IComponentFactory<TScalarTrainer> BasePredictorType; 48internal override IComponentFactory<TScalarTrainer> GetPredictorFactory() => BasePredictorType;

OutputCombiners\Stacking.cs (3)

18using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 45public IComponentFactory<TScalarTrainer> BasePredictorType; 47internal override IComponentFactory<TScalarTrainer> GetPredictorFactory() => BasePredictorType;

Trainer\Binary\EnsembleTrainer.cs (3)

26using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 52public IComponentFactory<TScalarTrainer>[] BasePredictors; 54internal override IComponentFactory<TScalarTrainer>[] GetPredictorFactories() => BasePredictors;

Trainer\EnsembleTrainerBase.cs (3)

53internal abstract IComponentFactory<ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<TOutput>>, IPredictorProducing<TOutput>>>[] GetPredictorFactories(); 64private protected readonly ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<TOutput>>, IPredictorProducing<TOutput>>[] Trainers; 97Trainers = new ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<TOutput>>, IPredictorProducing<TOutput>>[NumModels];

Trainer\Multiclass\MulticlassDataPartitionEnsembleTrainer.cs (3)

27using TVectorTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<VBuffer<float>>>, IPredictorProducing<VBuffer<float>>>; 53internal IComponentFactory<TVectorTrainer>[] BasePredictors; 55internal override IComponentFactory<TVectorTrainer>[] GetPredictorFactories() => BasePredictors;

Trainer\Regression\RegressionEnsembleTrainer.cs (3)

26using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 47public IComponentFactory<TScalarTrainer>[] BasePredictors; 49internal override IComponentFactory<TScalarTrainer>[] GetPredictorFactories() => BasePredictors;

Helpers\EnvironmentFactory.cs (2)

17where TTrainer : ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel> 33where TTrainer : ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel>

Standard\MulticlassClassification\MetaMulticlassTrainer.cs (5)

15using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 25internal IComponentFactory<TScalarTrainer> PredictorType; 45private protected readonly TScalarTrainer Trainer; 64TScalarTrainer singleEstimator = null, ICalibratorTrainer calibrator = null) 84private TScalarTrainer CreateTrainer()

Standard\MulticlassClassification\OneVersusAllTrainer.cs (5)

40using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 86/// <seealso cref="StandardTrainersCatalog.OneVersusAll{TModel}(MulticlassClassificationCatalog.MulticlassClassificationTrainers, ITrainerEstimator{BinaryPredictionTransformer{TModel}, TModel}, string, bool, IEstimator{ISingleFeaturePredictionTransformer{ICalibrator}}, int, bool)" /> 125/// <param name="binaryEstimator">An instance of a binary <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> used as the base trainer.</param> 132TScalarTrainer binaryEstimator, 163private ISingleFeaturePredictionTransformer<TScalarPredictor> TrainOne(IChannel ch, TScalarTrainer trainer, RoleMappedData data, int cls)

Standard\MulticlassClassification\PairwiseCouplingTrainer.cs (5)

28using TScalarTrainer = ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>>; 75/// <seealso cref="StandardTrainersCatalog.PairwiseCoupling{TModel}(MulticlassClassificationCatalog.MulticlassClassificationTrainers, ITrainerEstimator{ISingleFeaturePredictionTransformer{TModel}, TModel}, string, bool, IEstimator{ISingleFeaturePredictionTransformer{ICalibrator}}, int)" /> 104/// <param name="binaryEstimator">An instance of a binary <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> used as the base trainer.</param> 110TScalarTrainer binaryEstimator, 145private ISingleFeaturePredictionTransformer<TDistPredictor> TrainOne(IChannel ch, TScalarTrainer trainer, RoleMappedData data, int cls1, int cls2)

StandardTrainersCatalog.cs (6)

739/// <param name="binaryEstimator">An instance of a binary <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> used as the base trainer.</param> 753ITrainerEstimator<BinaryPredictionTransformer<TModel>, TModel> binaryEstimator, 763if (!(binaryEstimator is ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>> est)) 780/// <param name="binaryEstimator">An instance of a binary <see cref="ITrainerEstimator{TTransformer, TPredictor}"/> used as the base trainer.</param> 793ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel> binaryEstimator, 802if (!(binaryEstimator is ITrainerEstimator<ISingleFeaturePredictionTransformer<IPredictorProducing<float>>, IPredictorProducing<float>> est))

FeatureContributionTests.cs (2)

192ITrainerEstimator<ISingleFeaturePredictionTransformer<ICalculateFeatureContribution>, ICalculateFeatureContribution> trainer, 214ITrainerEstimator<ISingleFeaturePredictionTransformer<CalibratedModelParametersBase<TModelParameters, TCalibrator>>, CalibratedModelParametersBase<TModelParameters, TCalibrator>> trainer,