IPredictor

Prediction\IPredictor.cs (1)

51internal interface IPredictorProducing<out TResult> : IPredictor

Dirty\PredictorInterfaces.cs (1)

175internal interface ICanGetTrainingLabelNames : IPredictor

Prediction\ICalculateFeatureContribution.cs (1)

14internal interface IFeatureContributionMapper : IPredictor

Scorers\FeatureContributionCalculation.cs (1)

92private sealed class BindableMapper : ISchemaBindableMapper, ICanSaveModel, IPredictor

MatrixFactorizationPredictor.cs (1)

35public sealed class MatrixFactorizationModelParameters : IPredictor, ICanSaveModel, ICanSaveInTextFormat, ISchemaBindableMapper

EntryPoints\PredictorModel.cs (2)

38internal abstract IPredictor Predictor { get; } 51internal abstract void PrepareData(IHostEnvironment env, IDataView input, out RoleMappedData roleMappedData, out IPredictor predictor);

Prediction\IPredictor.cs (1)

12/// <see cref="ITrainer"/> and <see cref="IPredictor"/> it is still useful, but for things based on

Prediction\ITrainer.cs (4)

62IPredictor Train(TrainContext context); 72where TPredictor : IPredictor 93public static IPredictor Train(this ITrainer trainer, RoleMappedData trainData) 104public static TPredictor Train<TPredictor>(this ITrainer<TPredictor> trainer, RoleMappedData trainData) where TPredictor : IPredictor

Prediction\TrainContext.cs (2)

44public IPredictor InitialPredictor { get; } 53public TrainContext(RoleMappedData trainingSet, RoleMappedData validationSet = null, RoleMappedData testSet = null, IPredictor initialPredictor = null)

UnitTests\TestEntryPoints.cs (1)

2186var predictor = trainer.Train(rmd);

Commands\CrossValidationCommand.cs (4)

149IPredictor inputPredictor = null; 358private readonly IPredictor _inputPredictor; 396IPredictor inputPredictor = null, 523var predictor = TrainUtils.Train(host, ch, trainData, trainer, validData,

Commands\DataCommand.cs (1)

248bool? wantPredictor, out IPredictor predictor,

Commands\SavePredictorCommand.cs (4)

131IPredictor predictor; 141public static void SavePredictor(IChannel ch, IPredictor predictor, RoleMappedSchema schema, 207public static void LoadModel(IHostEnvironment env, Stream modelStream, bool loadNames, out IPredictor predictor, out RoleMappedSchema schema) 215ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(env, out predictor, rep, ModelFileUtils.DirPredictor);

Commands\ScoreCommand.cs (10)

40internal delegate void SignatureBindableMapper(IPredictor predictor); 104LoadModelObjects(ch, true, out var predictor, true, out var trainSchema, out var loader); 241public static IDataScorerTransform GetScorer(IPredictor predictor, RoleMappedData data, IHostEnvironment env, RoleMappedSchema trainSchema) 249IPredictor predictor, 256IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory = null) 276IPredictor predictor, 280IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory, 364IPredictor predictor, 365IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory = null, 393private static bool TryCreateBindableFromScorer(IHostEnvironment env, IPredictor predictor,

Commands\TestCommand.cs (1)

92IPredictor predictor;

Commands\TrainCommand.cs (12)

144IPredictor inputPredictor = null; 200var predictor = TrainUtils.Train(Host, ch, data, trainer, validData, 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, 253IComponentFactory<ICalibratorTrainer> calibrator, int maxCalibrationExamples, bool? cacheData, IPredictor inputPredictor = null, RoleMappedData testData = null) 258private static IPredictor TrainCore(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, RoleMappedData validData, 259IComponentFactory<ICalibratorTrainer> calibrator, int maxCalibrationExamples, bool? cacheData, IPredictor inputPredictor = null, RoleMappedData testData = null) 280var predictor = trainer.Train(new TrainContext(data, validData, testData, inputPredictor)); 285public static bool TryLoadPredictor(IChannel ch, IHostEnvironment env, string inputModelFile, out IPredictor inputPredictor) 298return ModelLoadContext.LoadModelOrNull<IPredictor, SignatureLoadModel>(env, out inputPredictor, rep, ModelFileUtils.DirPredictor); 318IPredictor predictor, RoleMappedData data, string command = null) 342public static void SaveModel(IHostEnvironment env, IChannel ch, Stream outputStream, IPredictor predictor, RoleMappedData data, string command = null)

Commands\TrainTestCommand.cs (2)

127IPredictor inputPredictor = null; 186var predictor = TrainUtils.Train(Host, ch, data, trainer, validData,

DataLoadSave\TransformerChain.cs (1)

292var predictor = ModelFileUtils.LoadPredictorOrNull(env, stream);

DataView\DataViewConstructionUtils.cs (1)

128var predictor = env.LoadPredictorOrNull(modelStream);

Dirty\ModelParametersBase.cs (1)

69PredictionKind IPredictor.PredictionKind => PredictionKind;

Dirty\PredictorUtils.cs (5)

18public static void SaveSummary(IChannel ch, IPredictor predictor, RoleMappedSchema schema, TextWriter writer) 38public static void SaveText(IChannel ch, IPredictor predictor, RoleMappedSchema schema, TextWriter writer) 70public static void SaveBinary(IChannel ch, IPredictor predictor, BinaryWriter writer) 85public static void SaveIni(IChannel ch, IPredictor predictor, RoleMappedSchema schema, TextWriter writer) 117public static void SaveCode(IChannel ch, IPredictor predictor, RoleMappedSchema schema, TextWriter writer)

EntryPoints\InputBase.cs (1)

119var predictor = TrainUtils.Train(host, ch, cachedRoleMappedData, trainer, calibrator, maxCalibrationExamples);

EntryPoints\PredictorModelImpl.cs (5)

27internal override IPredictor Predictor { get; } 30internal PredictorModelImpl(IHostEnvironment env, RoleMappedData trainingData, IDataView startingData, IPredictor predictor) 60private PredictorModelImpl(TransformModel transformModel, IPredictor predictor, KeyValuePair<RoleMappedSchema.ColumnRole, string>[] roleMappings) 99internal override void PrepareData(IHostEnvironment env, IDataView input, out RoleMappedData roleMappedData, out IPredictor predictor) 112var predictor = Predictor;

EntryPoints\SummarizePredictor.cs (2)

40IPredictor predictor; 49internal static IDataView GetSummaryAndStats(IHostEnvironment env, IPredictor predictor, RoleMappedSchema schema, out IDataView stats)

Model\Pfa\SavePfaCommand.cs (1)

124IPredictor rawPred;

Prediction\Calibrator.cs (10)

138IPredictor Calibrate(IChannel ch, IDataView data, ICalibratorTrainer caliTrainer, int maxRows); 209PredictionKind IPredictor.PredictionKind => ((IPredictorProducing<float>)SubModel).PredictionKind; 845ITrainer trainer, IPredictor predictor, RoleMappedSchema schema) 907public static IPredictor TrainCalibratorIfNeeded(IHostEnvironment env, IChannel ch, ICalibratorTrainer calibrator, 908int maxRows, ITrainer trainer, IPredictor predictor, RoleMappedData data) 933public static IPredictor GetCalibratedPredictor(IHostEnvironment env, IChannel ch, ICalibratorTrainer caliTrainer, 934IPredictor predictor, RoleMappedData data, int maxRows = _maxCalibrationExamples) 1013public static ICalibrator TrainCalibrator(IHostEnvironment env, IChannel ch, ICalibratorTrainer caliTrainer, IPredictor predictor, RoleMappedData data, int maxRows = _maxCalibrationExamples) 2214IPredictor predictor; 2220IPredictor calibratedPredictor;

Prediction\IModelCombiner.cs (2)

18IPredictor CombineModels(IEnumerable<IPredictor> models);

Prediction\IPredictionTransformer.cs (3)

11/// An interface for all the transformer that can transform data based on the <see cref="IPredictor"/> field. 15/// <typeparam name="TModel">The <see cref="IPredictor"/> or <see cref="ICalibrator"/> used for the data transformation.</typeparam> 27/// <typeparam name="TModel">The <see cref="IPredictor"/> or <see cref="ICalibrator"/> used for the data transformation.</typeparam>

Scorers\FeatureContributionCalculation.cs (1)

75private static ISchemaBindableMapper Create(IHostEnvironment env, Arguments args, IPredictor predictor)

Scorers\PredictionTransformer.cs (3)

51private protected IPredictor ModelAsPredictor => (IPredictor)Model; 87Host.CheckParam(model is IPredictor, nameof(model));

Scorers\QuantileRegressionScorer.cs (1)

40private static ISchemaBindableMapper Create(IHostEnvironment env, Arguments args, IPredictor predictor)

Scorers\SchemaBindablePredictorWrapper.cs (10)

32/// This is a base class for wrapping <see cref="IPredictor"/>s in an <see cref="ISchemaBindableMapper"/>. 43protected readonly IPredictor Predictor; 51public SchemaBindablePredictorWrapperBase(IPredictor predictor) 59private static DataViewType GetScoreType(IPredictor predictor, out IValueMapper valueMapper) 77ctx.LoadModel<IPredictor, SignatureLoadModel>(env, out Predictor, ModelFileUtils.DirPredictor); 277/// This class is a wrapper for all <see cref="IPredictor"/>s except for quantile regression predictors, 297public SchemaBindablePredictorWrapper(IPredictor predictor) 358private static string GetScoreColumnKind(IPredictor predictor) 405public SchemaBindableBinaryPredictorWrapper(IPredictor predictor) 642public SchemaBindableQuantileRegressionPredictor(IPredictor predictor, double[] quantiles)

Training\TrainerEstimatorBase.cs (8)

16public abstract class TrainerEstimatorBase<TTransformer, TModel> : ITrainerEstimator<TTransformer, TModel>, ITrainer<IPredictor> 97IPredictor ITrainer<IPredictor>.Train(TrainContext context) 100var pred = TrainModelCore(context) as IPredictor; 145IDataView validationSet = null, IPredictor initPredictor = null) 168IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);

Training\TrainerUtils.cs (3)

400where TPredictor : IPredictor 455IPredictor ITrainer.Train(TrainContext context) => Train(context); 473where TPredictor : IPredictor

Transforms\TrainAndScoreTransformer.cs (5)

83IPredictor predictor; 89ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(env, out predictor, rep, ModelFileUtils.DirPredictor); 214internal static IDataTransform Create(IHostEnvironment env, Arguments args, IDataView input, IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory) 226private static IDataTransform Create(IHostEnvironment env, Arguments args, ITrainer trainer, IDataView input, IComponentFactory<IPredictor, ISchemaBindableMapper> mapperFactory) 242var predictor = TrainUtils.Train(host, ch, data, trainer, null,

Utilities\ComponentCreation.cs (3)

156RoleMappedData data, IPredictor predictor, RoleMappedSchema trainSchema = null) 187IPredictor predictor, RoleMappedSchema trainSchema = null) 209public static IPredictor LoadPredictorOrNull(this IHostEnvironment env, Stream modelStream)

Utilities\ModelFileUtils.cs (4)

143public static IPredictor LoadPredictorOrNull(IHostEnvironment env, Stream modelStream) 154public static IPredictor LoadPredictorOrNull(IHostEnvironment env, RepositoryReader rep) 158IPredictor predictor; 159ModelLoadContext.LoadModelOrNull<IPredictor, SignatureLoadModel>(env, out predictor, rep, DirPredictor);

EntryPoints\CreateEnsemble.cs (3)

106model.PrepareData(env, inputData, out RoleMappedData transformedDataCur, out IPredictor pred); 158var ensemble = trainer.CombineModels(input.Models.Select(pm => pm.Predictor as IPredictorProducing<float>)); 190var ensemble = trainer.CombineModels(input.Models.Select(pm => pm.Predictor as IPredictorProducing<float>));

EntryPoints\PipelineEnsemble.cs (2)

36out RoleMappedData rmd, out IPredictor predictor); 53pm.PrepareData(host, new EmptyDataView(host, pm.TransformModel.InputSchema), out rmd, out IPredictor pred);

PipelineEnsemble.cs (7)

68Parent.PredictorModels[i].PrepareData(Parent.Host, emptyDv, out RoleMappedData rmd, out IPredictor predictor); 322public IPredictor Calibrate(IChannel ch, IDataView data, ICalibratorTrainer caliTrainer, int maxRows) 583PredictorModels[i].PrepareData(Host, dv, out RoleMappedData rmd, out IPredictor pred); 596model.PrepareData(env, edv, out RoleMappedData rmd, out IPredictor pred); 618private static int CheckNonKeyLabelColumnCore(IHostEnvironment env, IPredictor pred, PredictorModel[] models, bool isBinary, DataViewType labelType) 659model.PrepareData(env, edv, out RoleMappedData rmd, out IPredictor pred); 721PredictorModels[i].PrepareData(Host, dv, out RoleMappedData rmd, out IPredictor pred);

Trainer\Binary\EnsembleTrainer.cs (4)

83private protected override IPredictor CreatePredictor(List<FeatureSubsetModel<float>> models) 90public IPredictor CombineModels(IEnumerable<IPredictor> models) 95var p = models.First();

Trainer\EnsembleModelParametersBase.cs (2)

68ctx.LoadModel<IPredictor, SignatureLoadModel>(Host, out IPredictor p, string.Format(SubPredictorFmt, i));

Trainer\EnsembleTrainerBase.cs (7)

20internal abstract class EnsembleTrainerBase<TOutput, TSelector, TCombiner> : ITrainer<IPredictor> 108IPredictor ITrainer<IPredictor>.Train(TrainContext context) 118IPredictor ITrainer.Train(TrainContext context) 119=> ((ITrainer<IPredictor>)this).Train(context); 121private IPredictor TrainCore(IChannel ch, RoleMappedData data) 193private protected abstract IPredictor CreatePredictor(List<FeatureSubsetModel<TOutput>> models);

Trainer\Multiclass\MulticlassDataPartitionEnsembleTrainer.cs (3)

90private protected override IPredictor CreatePredictor(List<FeatureSubsetModel<VBuffer<float>>> models) 95public IPredictor CombineModels(IEnumerable<IPredictor> models)

Trainer\Regression\RegressionEnsembleTrainer.cs (4)

78private protected override IPredictor CreatePredictor(List<FeatureSubsetModel<float>> models) 83public IPredictor CombineModels(IEnumerable<IPredictor> models) 89var p = models.First();

PermutationFeatureImportance.cs (6)

29input.PredictorModel.PrepareData(env, input.Data, out RoleMappedData roleMappedData, out IPredictor predictor); 61IPredictor predictor, 86IPredictor predictor, 143IPredictor predictor, 202IPredictor predictor, 253IPredictor predictor,

ScoreModel.cs (2)

76input.PredictorModel.PrepareData(host, inputData, out RoleMappedData data, out IPredictor predictor); 123IPredictor predictor;

GamModelParameters.cs (1)

872IPredictor rawPred;

TreeEnsemble\TreeEnsembleCombiner.cs (4)

37IPredictor IModelCombiner.CombineModels(IEnumerable<IPredictor> models) 45foreach (var model in models) 49var predictor = model;

TreeEnsembleFeaturizer.cs (7)

397public TreeEnsembleFeaturizerBindableMapper(IHostEnvironment env, Arguments args, IPredictor predictor) 596TreeEnsembleFeaturizerBindableMapper.Arguments args, IPredictor predictor) 639IPredictor predictor; 642ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(host, out predictor, rep, ModelFileUtils.DirPredictor); 679var mapperFactory = ComponentFactoryUtils.CreateFromFunction<IPredictor, ISchemaBindableMapper>( 711var predictor = args.PredictorModel.Predictor; 714args.PredictorModel.PrepareData(env, input, out data, out var predictor2);

SymSgdClassificationTrainer.cs (1)

222var initPred = context.InitialPredictor;

SaveOnnxCommand.cs (1)

294IPredictor rawPred = null;

TestPredictors.cs (2)

671var fastTree = combiner.CombineModels(fastTrees.Select(pm => (IPredictorProducing<float>)pm.Predictor)); 841var predictor = combiner.CombineModels(predictors.Select(pm => pm.Predictor));

MatrixFactorizationPredictor.cs (1)

85PredictionKind IPredictor.PredictionKind => PredictionKind.Recommendation;

MatrixFactorizationTrainer.cs (1)

432IPredictor ITrainer.Train(TrainContext context) => Train(context);

FactorizationMachine\FactorizationMachineTrainer.cs (1)

567IPredictor ITrainer.Train(TrainContext context) => Train(context);

Standard\LinearPredictorUtils.cs (1)

96public static string LinearModelAsIni(in VBuffer<float> weights, float bias, IPredictor predictor = null,

Standard\LogisticRegression\LbfgsPredictorBase.cs (3)

164private IPredictor _srcPredictor; 417private protected abstract VBuffer<float> InitializeWeightsFromPredictor(IPredictor srcPredictor); 433Host.CheckParam(context.InitialPredictor == null || context.InitialPredictor is IPredictor, nameof(context.InitialPredictor));

Standard\LogisticRegression\LogisticRegression.cs (1)

437private protected override VBuffer<float> InitializeWeightsFromPredictor(IPredictor srcPredictor)

Standard\LogisticRegression\MulticlassLogisticRegression.cs (1)

280private protected override VBuffer<float> InitializeWeightsFromPredictor(IPredictor srcPredictor)

Standard\MulticlassClassification\MetaMulticlassTrainer.cs (7)

17public abstract class MetaMulticlassTrainer<TTransformer, TModel> : ITrainerEstimator<TTransformer, TModel>, ITrainer<IPredictor> 117IPredictor ITrainer<IPredictor>.Train(TrainContext context) 130var pred = TrainCore(ch, data, count) as IPredictor; 184IPredictor ITrainer.Train(TrainContext context) => ((ITrainer<IPredictor>)this).Train(context);

Standard\Online\OnlineLinear.cs (1)

277var initPredictor = context.InitialPredictor;

Standard\PoissonRegression\PoissonRegression.cs (1)

137private protected override VBuffer<float> InitializeWeightsFromPredictor(IPredictor srcPredictor)

Standard\SdcaBinary.cs (1)

75var initPred = context.InitialPredictor;

Standard\Simple\SimpleTrainers.cs (1)

296IPredictor ITrainer.Train(TrainContext context) => Train(context);

Standard\StochasticTrainerBase.cs (1)

36var initPred = context.InitialPredictor;

Scenarios\Api\Estimators\TrainWithInitialPredictor.cs (1)

44var finalModel = ((ITrainer)secondTrainer).Train(new TrainContext(trainRoles, initialPredictor: firstModel.Model));

LearnerFeatureSelection.cs (1)

294var predictor = TrainUtils.Train(host, ch, data, trainer, null,