ModelFileUtils

UnitTests\TestModelLoad.cs (2)

30ILegacyDataLoader result = ModelFileUtils.LoadLoader(env, rep, new MultiFileSource(null), true); 48var result = ModelFileUtils.LoadPipeline(env, rep, new MultiFileSource(null), true);

Commands\DataCommand.cs (7)

206return ModelFileUtils.LoadLoader(Host, rep, new MultiFileSource(path), loadTransforms); 265predictor = ModelFileUtils.LoadPredictorOrNull(Host, rep); 299var trainRoleMappings = ModelFileUtils.LoadRoleMappingsOrNull(Host, rep); 307trainPipe = ModelFileUtils.LoadLoader(Host, rep, new MultiFileSource(null), loadTransforms: true); 388using (var pipeLoaderEntry = rep.OpenEntry(ModelFileUtils.DirDataLoaderModel, ModelLoadContext.ModelStreamName)) 389using (var ctx = new ModelLoadContext(rep, pipeLoaderEntry, ModelFileUtils.DirDataLoaderModel)) 424ModelSaveContext.SaveModel(rep, loader, ModelFileUtils.DirDataLoaderModel);

Commands\SavePredictorCommand.cs (4)

215ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(env, out predictor, rep, ModelFileUtils.DirPredictor); 219var roles = ModelFileUtils.LoadRoleMappingsOrNull(env, rep); 222var emptyView = ModelFileUtils.LoadPipeline(env, rep, new MultiFileSource(null)); 228if (ModelFileUtils.TryLoadFeatureNames(out names, rep))

Commands\TrainCommand.cs (6)

298return ModelLoadContext.LoadModelOrNull<IPredictor, SignatureLoadModel>(env, out inputPredictor, rep, ModelFileUtils.DirPredictor); 359ModelSaveContext.SaveModel(rep, predictor, ModelFileUtils.DirPredictor); 365ModelSaveContext.SaveModel(rep, dataPipe, ModelFileUtils.DirDataLoaderModel); 373using (var ent = rep.CreateEntry(ModelFileUtils.DirTrainingInfo, "Command.txt")) 377ModelFileUtils.SaveRoleMappings(env, ch, data.Schema, rep); 407using (var ctx = ModelFileUtils.GetDataModelSavingContext(repositoryWriter))

DataLoadSave\TransformerChain.cs (3)

290var chain = ModelFileUtils.LoadPipeline(env, stream, new MultiFileSource(null), extractInnerPipe: false); 292var predictor = ModelFileUtils.LoadPredictorOrNull(env, stream); 295var roles = ModelFileUtils.LoadRoleMappingsOrNull(env, stream);

DataView\DataViewConstructionUtils.cs (1)

131var roles = ModelFileUtils.LoadRoleMappingsOrNull(env, modelStream);

EntryPoints\PredictorModelImpl.cs (2)

51var roles = ModelFileUtils.LoadRoleMappingsOrNull(env, stream); 55Predictor = ModelFileUtils.LoadPredictorOrNull(env, stream);

EntryPoints\TransformModelImpl.cs (1)

116_chain = ModelFileUtils.LoadPipeline(env, stream, new MultiFileSource(null), extractInnerPipe: true);

Model\ModelOperationsCatalog.cs (1)

182var loader = ModelFileUtils.LoadLoader(_env, rep, new MultiFileSource(null), false);

Prediction\Calibrator.cs (2)

269ctx.SaveModel(SubModel, ModelFileUtils.DirPredictor); 276ctx.LoadModel<TSubModel, SignatureLoadModel>(env, out predictor, ModelFileUtils.DirPredictor);

Scorers\FeatureContributionCalculation.cs (2)

152ctx.LoadModel<IFeatureContributionMapper, SignatureLoadModel>(env, out Predictor, ModelFileUtils.DirPredictor); 182ctx.SaveModel(Predictor, ModelFileUtils.DirPredictor);

Scorers\SchemaBindablePredictorWrapper.cs (2)

77ctx.LoadModel<IPredictor, SignatureLoadModel>(env, out Predictor, ModelFileUtils.DirPredictor); 91ctx.SaveModel(Predictor, ModelFileUtils.DirPredictor);

Transforms\FeatureContributionCalculationTransformer.cs (2)

121ctx.LoadModel<IFeatureContributionMapper, SignatureLoadModel>(env, out _predictor, ModelFileUtils.DirPredictor); 142ctx.SaveModel(_predictor, ModelFileUtils.DirPredictor);

Transforms\TrainAndScoreTransformer.cs (2)

89ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(env, out predictor, rep, ModelFileUtils.DirPredictor); 90trainSchema = ModelFileUtils.LoadRoleMappedSchemaOrNull(env, rep);

Utilities\ComponentCreation.cs (2)

74return ModelFileUtils.LoadTransforms(env, data, modelStream); 212return ModelFileUtils.LoadPredictorOrNull(env, modelStream);

Utilities\ModelFileUtils.cs (4)

212using (var ent = rep.OpenEntryOrNull(ModelFileUtils.DirTrainingInfo, "FeatureNames.bin")) 216using (var ctx = new ModelLoadContext(rep, ent, ModelFileUtils.DirTrainingInfo)) 347var roleMappings = ModelFileUtils.LoadRoleMappingsOrNull(env, rep); 350var pipe = ModelFileUtils.LoadLoader(h, rep, new MultiFileSource(null), loadTransforms: true);

TreeEnsembleFeaturizer.cs (1)

642ModelLoadContext.LoadModel<IPredictor, SignatureLoadModel>(host, out predictor, rep, ModelFileUtils.DirPredictor);

DnnImageFeaturizerTest.cs (1)

161var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

OnnxTransformTests.cs (1)

204var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

PartitionedFileLoader.cs (1)

361return ModelFileUtils.LoadLoader(_host, rep, files, false);

ResultProcessor.cs (1)

470var ent = rep.OpenEntryOrNull(ModelFileUtils.DirTrainingInfo, "Command.txt");

TensorFlowEstimatorTests.cs (1)

132var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

TestCommandBase.cs (1)

542out pipe, rep, ModelFileUtils.DirDataLoaderModel, files);

ImagesTests.cs (10)

TermEstimatorTests.cs (1)

129var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

TextLoaderTests.cs (2)

820var result = ModelFileUtils.LoadLoader(mlContext, rep, new MultiFileSource(breastCancerPath), false); 840iris = ModelFileUtils.LoadLoader(mlContext, rep, new MultiFileSource(irisPath), false);

TrainerEstimators\LbfgsTests.cs (2)

148var result = ModelFileUtils.LoadPredictorOrNull(Env, fs) as CalibratedModelParametersBase<LinearBinaryModelParameters, PlattCalibrator>; 239var result = ModelFileUtils.LoadPredictorOrNull(Env, fs) as MaximumEntropyModelParameters;

Transformers\CategoricalHashTests.cs (1)

237var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\CategoricalTests.cs (1)

330var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\CharTokenizeTests.cs (1)

69var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\ConvertTests.cs (1)

329var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\CopyColumnEstimatorTests.cs (1)

125var loadedView = ModelFileUtils.LoadTransforms(env, dataView, ms);

Transformers\FeatureSelectionTests.cs (2)

161var loadedView = ModelFileUtils.LoadTransforms(ML, dataView, ms); 221var loadedView = ModelFileUtils.LoadTransforms(ML, dataView, ms);

Transformers\HashTests.cs (1)

125var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\KeyToBinaryVectorEstimatorTest.cs (1)

162var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\KeyToVectorEstimatorTests.cs (1)

224var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\NAIndicatorTests.cs (1)

87var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\NAReplaceTests.cs (1)

182var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\NormalizerTests.cs (4)

745var loadedView = ModelFileUtils.LoadTransforms(ML, dataView, ms); 806var loadedView = ModelFileUtils.LoadTransforms(ML, dataView, ms); 866var loadedView = ModelFileUtils.LoadTransforms(ML, dataView, ms); 878var result = ModelFileUtils.LoadTransforms(Env, dataView, fs);

Transformers\RffTests.cs (1)

110var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\SelectColumnsTests.cs (4)

223var result = ModelFileUtils.LoadTransforms(Env, dataView, fs); 251var result = ModelFileUtils.LoadTransforms(Env, dataView, fs); 279var result = ModelFileUtils.LoadTransforms(Env, dataView, fs); 307var result = ModelFileUtils.LoadTransforms(Env, dataView, fs);

Transformers\TextFeaturizerTests.cs (1)

662var result = ModelFileUtils.LoadTransforms(Env, data, fs);

Transformers\TextNormalizer.cs (1)

90var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

Transformers\ValueMappingTests.cs (2)

719var result = ModelFileUtils.LoadTransforms(Env, dataView, fs); 736var result = ModelFileUtils.LoadTransforms(Env, dataView, fs);

Transformers\WordTokenizeTests.cs (1)

110var loadedView = ModelFileUtils.LoadTransforms(Env, dataView, ms);

LoadTransform.cs (2)

101using (var pipeLoaderEntry = rep.OpenEntry(ModelFileUtils.DirDataLoaderModel, ModelLoadContext.ModelStreamName)) 102using (var ctx = new ModelLoadContext(rep, pipeLoaderEntry, ModelFileUtils.DirDataLoaderModel))

Text\SentimentAnalyzingTransform.cs (1)

153view = ModelFileUtils.LoadTransforms(env, view, rep);