3 instantiations of LightGbmBinaryModelParameters
Microsoft.ML.LightGbm (3)
LightGbmBinaryTrainer.cs (2)
80var predictor = new LightGbmBinaryModelParameters(env, ctx); 259var pred = new LightGbmBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, innerArgs);
LightGbmMulticlassTrainer.cs (1)
208return new LightGbmBinaryModelParameters(Host, GetBinaryEnsemble(classID), FeatureCount, innerArgs);
33 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); 87return new ValueMapperCalibratedModelParameters<LightGbmBinaryModelParameters, ICalibrator>(env, predictor, calibrator); 117BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>, 118CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>> 255private protected override CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator> CreatePredictor() 259var pred = new LightGbmBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, innerArgs); 261return new FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>(Host, pred, cali); 294private protected override BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>> 295MakeTransformer(CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator> model, DataViewSchema trainSchema) 296=> new BinaryPredictionTransformer<CalibratedModelParametersBase<LightGbmBinaryModelParameters, PlattCalibrator>>(Host, model, trainSchema, FeatureColumn.Name); 302public 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 (9)
CalibratedModelParametersTests.cs (8)
97[InlineData(typeof(LightGbmBinaryModelParameters))] 114[InlineData(typeof(LightGbmBinaryModelParameters), false)] 122: modelType == typeof(LightGbmBinaryModelParameters) 145[InlineData(typeof(LightGbmBinaryModelParameters))] 164[InlineData(typeof(LightGbmBinaryModelParameters))] 207if (modelType == typeof(LightGbmBinaryModelParameters)) 220LightGbmBinaryModelParameters lightGbm => new FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>(Env, lightGbm, calibrator),
TrainerEstimators\TreeEstimators.cs (1)
425((FeatureWeightsCalibratedModelParameters<LightGbmBinaryModelParameters, PlattCalibrator>)predictor).Calibrator.Slope == -sigmoid));