public static void LoadModel(IHostEnvironment env, Stream modelStream, bool loadNames, out IPredictor predictor, out RoleMappedSchema schema)
{
Contracts.CheckValue(env, nameof(env));
env.CheckValue(modelStream, nameof(modelStream));
schema = null;
using (var rep = RepositoryReader.Open(modelStream, env))
{
ModelLoadContext.LoadModel <IPredictor, SignatureLoadModel>(env, out predictor, rep, ModelFileUtils.DirPredictor);
if (loadNames)
{
var roles = ModelFileUtils.LoadRoleMappingsOrNull(env, rep);
if (roles != null)
{
var emptyView = ModelFileUtils.LoadPipeline(env, rep, new MultiFileSource(null));
schema = new RoleMappedSchema(emptyView.Schema, roles, opt: true);
}
else
{
FeatureNameCollection names;
if (ModelFileUtils.TryLoadFeatureNames(out names, rep))
{
schema = names.Schema;
}
}
}
}
}
