diff --git a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs index 73235797e5..a0d9577bfd 100644 --- a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs +++ b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs @@ -200,7 +200,8 @@ internal TransformerChain(IHostEnvironment env, ModelLoadContext ctx) LastTransformer = null; } - public void SaveTo(IHostEnvironment env, Stream outputStream) + [BestFriend] + internal void SaveTo(IHostEnvironment env, Stream outputStream) { using (var ch = env.Start("Saving pipeline")) { diff --git a/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs b/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs index fcc0905bad..0443240e1d 100644 --- a/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs +++ b/src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs @@ -53,21 +53,5 @@ internal ExplainabilityTransforms(ModelOperationsCatalog owner) _env = owner._env; } } - - /// - /// Create a prediction engine for one-time prediction. - /// - /// The class that defines the input data. - /// The class that defines the output data. - /// The transformer to use for prediction. - /// Additional settings of the input schema. - /// Additional settings of the output schema. - public PredictionEngine CreatePredictionEngine(ITransformer transformer, - SchemaDefinition inputSchemaDefinition = null, SchemaDefinition outputSchemaDefinition = null) - where TSrc : class - where TDst : class, new() - { - return new PredictionEngine(_env, transformer, false, inputSchemaDefinition, outputSchemaDefinition); - } } }