private void SetScorer()
{
var schema = new RoleMappedSchema(TrainSchema, _trainLabelColumn, FeatureColumn);
var args = new MultiClassClassifierScorer.Arguments();
Scorer = new MultiClassClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema);
}
public MulticlassPredictionTransformer(IHostEnvironment env, TModel model, ISchema inputSchema, string featureColumn, string labelColumn)
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(MulticlassPredictionTransformer <TModel>)), model, inputSchema, featureColumn)
{
Host.CheckValueOrNull(labelColumn);
_trainLabelColumn = labelColumn;
var schema = new RoleMappedSchema(inputSchema, labelColumn, featureColumn);
var args = new MultiClassClassifierScorer.Arguments();
_scorer = new MultiClassClassifierScorer(Host, args, new EmptyDataView(Host, inputSchema), BindableMapper.Bind(Host, schema), schema);
}
public MulticlassPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(MulticlassPredictionTransformer <TModel>)), ctx)
{
// *** Binary format ***
// <base info>
// id of string: train label column
_trainLabelColumn = ctx.LoadStringOrNull();
var schema = new RoleMappedSchema(TrainSchema, _trainLabelColumn, FeatureColumn);
var args = new MultiClassClassifierScorer.Arguments();
_scorer = new MultiClassClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema);
}
private MultiClassClassifierScorer(IHostEnvironment env, MultiClassClassifierScorer transform, IDataView newSource)
: base(env, transform, newSource, RegistrationName)
{
}