3 instantiations of LightGbmBinaryModelParameters
Microsoft.ML.LightGbm (3)
LightGbmBinaryTrainer.cs (2)
80var predictor = new LightGbmBinaryModelParameters(env, ctx); 257var pred = new LightGbmBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, innerArgs);
LightGbmMulticlassTrainer.cs (1)
208return new LightGbmBinaryModelParameters(Host, GetBinaryEnsemble(classID), FeatureCount, innerArgs);
25 references to LightGbmBinaryModelParameters
Microsoft.ML.AutoML (2)
TrainerExtensions\BinaryTrainerExtensions.cs (2)
118LightGbmBinaryTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions<LightGbmBinaryTrainer.Options, float, BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>, CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>(sweepParams, columnInfo);
Microsoft.ML.Core.Tests (4)
UnitTests\TestEntryPoints.cs (4)
327Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 2789Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 2796Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly); 6674Env.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly);
Microsoft.ML.LightGbm (17)
LightGbmBinaryTrainer.cs (14)
22[assembly: LoadableClass(typeof(IPredictorProducing<float>), typeof(LightGbmBinaryModelParameters), null, typeof(SignatureLoadModel), 24LightGbmBinaryModelParameters.LoaderSignature)] 51loaderAssemblyName: typeof(LightGbmBinaryModelParameters).Assembly.FullName); 80var predictor = new LightGbmBinaryModelParameters(env, ctx); 85return new ValueMapperCalibratedModelParameters<LightGbmBinaryModelParameters, ICalibrator>(env, predictor, calibrator); 115BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>, 116CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>> 253private protected override CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator> CreatePredictor() 257var pred = new LightGbmBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, innerArgs); 259return new FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>(Host, pred, cali); 292private protected override BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>> 293MakeTransformer(CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator> model, DataViewSchema trainSchema) 294=> new BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>(Host, model, trainSchema, FeatureColumn.Name); 300public BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>> Fit(IDataView trainData, IDataView validationData)
LightGbmMulticlassTrainer.cs (3)
206private LightGbmBinaryModelParameters CreateBinaryPredictor(int classID, string innerArgs) 222var pred = CreateBinaryPredictor(i, innerArgs); 224predictors[i] = new FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>(Host, pred, cali);
Microsoft.ML.Predictor.Tests (1)
TestPredictors.cs (1)
45environment.ComponentCatalog.RegisterAssembly(typeof(LightGbmBinaryModelParameters).Assembly);
Microsoft.ML.Tests (1)
TrainerEstimators\TreeEstimators.cs (1)
378((FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>)predictor).Calibrator.Slope == -sigmoid));