From 588116ff4024c894228999f1a98c2e51d3560c24 Mon Sep 17 00:00:00 2001 From: Ivan Matantsev Date: Mon, 11 Mar 2019 10:06:09 -0700 Subject: [PATCH 1/2] Keep hiding --- .../DataLoadSave/TransformerChain.cs | 2 +- .../Model/ModelOperationsCatalog.cs | 16 ---------------- 2 files changed, 1 insertion(+), 17 deletions(-) diff --git a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs index 73235797e5..e5cab81baa 100644 --- a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs +++ b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs @@ -200,7 +200,7 @@ internal TransformerChain(IHostEnvironment env, ModelLoadContext ctx) LastTransformer = null; } - public void SaveTo(IHostEnvironment env, Stream outputStream) + 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); - } } } From 63df62a098ca5fa50a3c30752ec56447b6fdd6a4 Mon Sep 17 00:00:00 2001 From: Ivan Matantsev Date: Mon, 11 Mar 2019 10:51:37 -0700 Subject: [PATCH 2/2] BestFriends! --- src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs | 1 + 1 file changed, 1 insertion(+) diff --git a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs index e5cab81baa..a0d9577bfd 100644 --- a/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs +++ b/src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs @@ -200,6 +200,7 @@ internal TransformerChain(IHostEnvironment env, ModelLoadContext ctx) LastTransformer = null; } + [BestFriend] internal void SaveTo(IHostEnvironment env, Stream outputStream) { using (var ch = env.Start("Saving pipeline"))