Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/Microsoft.ML.Data/DataLoadSave/TransformerChain.cs
Original file line number Diff line number Diff line change
Expand Up @@ -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"))
{
Expand Down
16 changes: 0 additions & 16 deletions src/Microsoft.ML.Data/Model/ModelOperationsCatalog.cs
Original file line number Diff line number Diff line change
Expand Up @@ -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);
}
}
}