private protected override IPredictorProducing <float> TrainModelCore(TrainContext context) { TrainBase(context); var predictor = new BinaryClassificationGamModelParameters(Host, InputLength, TrainSet, MeanEffect, BinEffects, FeatureMap); var calibrator = new PlattCalibrator(Host, -1.0 * _sigmoidParameter, 0); return(new CalibratedPredictor(Host, predictor, calibrator)); }
private static IPredictorProducing <float> Create(IHostEnvironment env, ModelLoadContext ctx) { Contracts.CheckValue(env, nameof(env)); env.CheckValue(ctx, nameof(ctx)); ctx.CheckAtModel(GetVersionInfo()); var predictor = new BinaryClassificationGamModelParameters(env, ctx); ICalibrator calibrator; ctx.LoadModelOrNull <ICalibrator, SignatureLoadModel>(env, out calibrator, @"Calibrator"); if (calibrator == null) { return(predictor); } return(new SchemaBindableCalibratedModelParameters <BinaryClassificationGamModelParameters, ICalibrator>(env, predictor, calibrator)); }