private void SetScorer()
{
var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumn);
var args = new BinaryClassifierScorer.Arguments {
Threshold = Threshold, ThresholdColumn = ThresholdColumn
};
Scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema);
}
public BinaryPredictionTransformer(IHostEnvironment env, TModel model, ISchema inputSchema, string featureColumn,
float threshold = 0f, string thresholdColumn = DefaultColumnNames.Score)
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(BinaryPredictionTransformer <TModel>)), model, inputSchema, featureColumn)
{
Host.CheckNonEmpty(thresholdColumn, nameof(thresholdColumn));
var schema = new RoleMappedSchema(inputSchema, null, featureColumn);
Threshold = threshold;
ThresholdColumn = thresholdColumn;
var args = new BinaryClassifierScorer.Arguments {
Threshold = Threshold, ThresholdColumn = ThresholdColumn
};
_scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, inputSchema), BindableMapper.Bind(Host, schema), schema);
}
public BinaryPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx)
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(BinaryPredictionTransformer <TModel>)), ctx)
{
// *** Binary format ***
// <base info>
// float: scorer threshold
// id of string: scorer threshold column
Threshold = ctx.Reader.ReadSingle();
ThresholdColumn = ctx.LoadString();
var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumn);
var args = new BinaryClassifierScorer.Arguments {
Threshold = Threshold, ThresholdColumn = ThresholdColumn
};
_scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema);
}
private BinaryClassifierScorer(IHostEnvironment env, BinaryClassifierScorer transform, IDataView newSource)
: base(env, transform, newSource, RegistrationName)
{
_threshold = transform._threshold;
}
