private protected override IPredictorWithFeatureWeights <float> CreatePredictor()
{
Host.Check(TrainedEnsemble != null, "The predictor cannot be created before training is complete");
var innerArgs = LightGbmInterfaceUtils.JoinParameters(Options);
var pred = new LightGbmBinaryPredictor(Host, TrainedEnsemble, FeatureCount, innerArgs);
var cali = new PlattCalibrator(Host, -0.5, 0);
return(new FeatureWeightsCalibratedPredictor(Host, pred, cali));
}
public static IPredictorProducing <float> Create(IHostEnvironment env, ModelLoadContext ctx)
{
Contracts.CheckValue(env, nameof(env));
env.CheckValue(ctx, nameof(ctx));
ctx.CheckAtModel(GetVersionInfo());
var predictor = new LightGbmBinaryPredictor(env, ctx);
ICalibrator calibrator;
ctx.LoadModelOrNull <ICalibrator, SignatureLoadModel>(env, out calibrator, @"Calibrator");
if (calibrator == null)
{
return(predictor);
}
return(new CalibratedPredictor(env, predictor, calibrator));
}
