DataLoadSave\LegacyCompositeDataLoader.cs (5)
293public static ILegacyDataLoader Create(IHostEnvironment env, ModelLoadContext ctx, IMultiStreamSource files)
321public static ILegacyDataLoader Create(IHostEnvironment env, ModelLoadContext ctx,
347public static IDataView LoadSelectedTransforms(ModelLoadContext ctx, IDataView srcView, IHostEnvironment env, Func<string, bool> isTransformTagAccepted)
450private static ILegacyDataLoader LoadTransforms(ModelLoadContext ctx, ILegacyDataLoader srcLoader, IHost host, Func<string, bool> isTransformTagAccepted)
554using (var loaderCtx = new ModelSaveContext(ctx.Repository, Path.Combine(ctx.Directory ?? "", "Loader"), ModelLoadContext.ModelStreamName))
Model\ModelOperationsCatalog.cs (4)
163ModelLoadContext.LoadModel<ITransformer, SignatureLoadModel>(_env, out var transformerChain, rep,
168ModelLoadContext.LoadModelOrNull<IDataLoader<IMultiStreamSource>, SignatureLoadModel>(_env, out var dataLoader, rep, null);
172if (ModelLoadContext.LoadModelOrNull<ITransformer, SignatureLoadModel>(_env, out var transformerChain, rep,
266ModelLoadContext.LoadModel<IDataLoader<IMultiStreamSource>, SignatureLoadModel>(_env, out loader, rep, null);
Prediction\Calibrator.cs (18)
273private protected static TSubModel GetPredictor(IHostEnvironment env, ModelLoadContext ctx)
280private protected static TCalibrator GetCalibrator(IHostEnvironment env, ModelLoadContext ctx)
435private ValueMapperCalibratedModelParameters(IHostEnvironment env, ModelLoadContext ctx, TSubModel predictor, TCalibrator calibrator)
440private static CalibratedModelParametersBase Create(IHostEnvironment env, ModelLoadContext ctx)
502private FeatureWeightsCalibratedModelParameters(IHostEnvironment env, ModelLoadContext ctx,
510private static CalibratedModelParametersBase Create(IHostEnvironment env, ModelLoadContext ctx)
579private ParameterMixingCalibratedModelParameters(IHostEnvironment env, ModelLoadContext ctx, TSubModel predictor, TCalibrator calibrator)
587private static CalibratedModelParametersBase Create(IHostEnvironment env, ModelLoadContext ctx)
759private SchemaBindableCalibratedModelParameters(IHostEnvironment env, ModelLoadContext ctx)
766internal static CalibratedModelParametersBase Create(IHostEnvironment env, ModelLoadContext ctx)
817internal static object Create(IHostEnvironment env, ModelLoadContext ctx, object predictor, ICalibrator calibrator, Type calibratedModelParametersType)
824typeof(ModelLoadContext),
1215private NaiveCalibrator(IHostEnvironment env, ModelLoadContext ctx)
1241internal static NaiveCalibrator Create(IHostEnvironment env, ModelLoadContext ctx)
1715private PlattCalibrator(IHostEnvironment env, ModelLoadContext ctx)
1731internal static PlattCalibrator Create(IHostEnvironment env, ModelLoadContext ctx)
1992private IsotonicCalibrator(IHostEnvironment env, ModelLoadContext ctx)
2036internal static IsotonicCalibrator Create(IHostEnvironment env, ModelLoadContext ctx)
Scorers\PredictionTransformer.cs (26)
95private protected PredictionTransformerBase(IHost host, ModelLoadContext ctx)
109private protected PredictionTransformerBase(IHost host, ModelLoadContext ctx, TModel model)
116private void InitializeLogic(IHost host, ModelLoadContext ctx)
241private protected SingleFeaturePredictionTransformerBase(IHost host, ModelLoadContext ctx)
256private protected SingleFeaturePredictionTransformerBase(IHost host, ModelLoadContext ctx, TModel model)
333internal AnomalyPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
414internal BinaryPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
420internal BinaryPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx, IHost host, TModel model)
426private void InitializationLogic(ModelLoadContext ctx, out float threshold, out string thresholdcolumn)
496internal MulticlassPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
502internal MulticlassPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx, IHost host, TModel model)
509private void InitializationLogic(ModelLoadContext ctx, out string trainLabelColumn, out string scoreColumn, out string predictedLabelColumn)
579internal RegressionPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
585internal RegressionPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx, IHost host, TModel model)
627internal RankingPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
633internal RankingPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx, IHost host, TModel model)
680internal ClusteringPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
719public static ISingleFeaturePredictionTransformer<object> Create(IHostEnvironment env, ModelLoadContext ctx)
736public static ISingleFeaturePredictionTransformer<object> Create(IHostEnvironment env, ModelLoadContext ctx)
753public static ISingleFeaturePredictionTransformer<object> Create(IHostEnvironment env, ModelLoadContext ctx)
771public static ISingleFeaturePredictionTransformer<object> Create(IHostEnvironment env, ModelLoadContext ctx)
785internal static object Create(IHostEnvironment env, ModelLoadContext ctx, IHost host, IPredictorProducing<float> model, Type predictionTransformerType)
796internal static object Create(IHostEnvironment env, ModelLoadContext ctx, IHost host, IPredictorProducing<VBuffer<float>> model, Type predictionTransformerType)
815typeof(ModelLoadContext),
861public static AnomalyPredictionTransformer<IPredictorProducing<float>> Create(IHostEnvironment env, ModelLoadContext ctx)
869public static ClusteringPredictionTransformer<IPredictorProducing<VBuffer<float>>> Create(IHostEnvironment env, ModelLoadContext ctx)
Transforms\NormalizeColumnDbl.cs (9)
88public static void LoadModel(ModelLoadContext ctx, ref List<int> indicesShift,
234public static void LoadModel(ModelLoadContext ctx, out TFloat[][] binUpperBounds)
290public static void LoadModel(ModelLoadContext ctx, int cv, out bool useLog, out TFloat[] mean, out TFloat[] stddev)
665public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
729public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
990public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
1055public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
1173public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
1258public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
Transforms\NormalizeColumnSng.cs (9)
88public static void LoadModel(ModelLoadContext ctx, ref List<int> indicesShift,
234public static void LoadModel(ModelLoadContext ctx, out TFloat[][] binUpperBounds)
290public static void LoadModel(ModelLoadContext ctx, int cv, out bool useLog, out TFloat[] mean, out TFloat[] stddev)
823public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
886public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
1150public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
1215public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
1334public static new ImplOne Create(ModelLoadContext ctx, IHost host, DataViewType typeSrc)
1419public static ImplVec Create(ModelLoadContext ctx, IHost host, VectorDataViewType typeSrc)
Utilities\ModelFileUtils.cs (11)
74ModelLoadContext.LoadModel<ITransformer, SignatureLoadModel>(env, out var transformerChain, rep, DirTransformerChain);
78using (var ent = rep.OpenEntry(DirDataLoaderModel, ModelLoadContext.ModelStreamName))
82ModelLoadContext.LoadModel<ILegacyDataLoader, SignatureLoadDataLoader>(env, out loader, rep, ent, DirDataLoaderModel, files);
131using (var ent = rep.OpenEntryOrNull(DirDataLoaderModel, ModelLoadContext.ModelStreamName))
135var ctx = new ModelLoadContext(rep, ent, DirDataLoaderModel);
159ModelLoadContext.LoadModelOrNull<IPredictor, SignatureLoadModel>(env, out predictor, rep, DirPredictor);
169return new ModelSaveContext(rep, DirDataLoaderModel, ModelLoadContext.ModelStreamName);
188ent = rep.OpenEntryOrNull(dir = Path.Combine(DirDataLoaderModel, "Loader"), ModelLoadContext.ModelStreamName);
191ent = rep.OpenEntry(dir = DirDataLoaderModel, ModelLoadContext.ModelStreamName);
198ModelLoadContext.LoadModel<ILegacyDataLoader, SignatureLoadDataLoader>(env, out loader, rep, ent, dir, files);
216using (var ctx = new ModelLoadContext(rep, ent, ModelFileUtils.DirTrainingInfo))