Hide some duplicate methods (#2904)

dotnet · Mar 12, 2019 · e2464f6 · e2464f6
1 parent 24f782d
commit e2464f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 17 deletions.
diff --git 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
@@ -53,21 +53,5 @@ internal ExplainabilityTransforms(ModelOperationsCatalog owner)
                 _env = owner._env;
             }
         }
-
-        /// <summary>
-        /// Create a prediction engine for one-time prediction.
-        /// </summary>
-        /// <typeparam name="TSrc">The class that defines the input data.</typeparam>
-        /// <typeparam name="TDst">The class that defines the output data.</typeparam>
-        /// <param name="transformer">The transformer to use for prediction.</param>
-        /// <param name="inputSchemaDefinition">Additional settings of the input schema.</param>
-        /// <param name="outputSchemaDefinition">Additional settings of the output schema.</param>
-        public PredictionEngine<TSrc, TDst> CreatePredictionEngine<TSrc, TDst>(ITransformer transformer,
-            SchemaDefinition inputSchemaDefinition = null, SchemaDefinition outputSchemaDefinition = null)
-            where TSrc : class
-            where TDst : class, new()
-        {
-            return new PredictionEngine<TSrc, TDst>(_env, transformer, false, inputSchemaDefinition, outputSchemaDefinition);
-        }
     }
 }