6 references to LightGbmRegression
Microsoft.ML.AutoML (3)
TrainerExtensions\TrainerExtensionCatalog.cs (1)
27
{ TrainerName.
LightGbmRegression
, typeof(LightGbmRegressionExtension) },
TrainerExtensions\TrainerExtensionUtil.cs (2)
199
trainerName == TrainerName.
LightGbmRegression
|| trainerName == TrainerName.LightGbmRanking)
354
return TrainerName.
LightGbmRegression
;
Microsoft.ML.AutoML.Tests (1)
TrainerExtensionsTests.cs (1)
327
var regressionParams = TrainerExtensionUtil.BuildParameterSet(TrainerName.
LightGbmRegression
, props);
Microsoft.ML.CodeGenerator (2)
CodeGenerator\CSharp\CodeGenerator.cs (1)
29
TrainerName.
LightGbmRegression
.ToString(), TrainerName.LightGbmRanking.ToString() };
CodeGenerator\CSharp\TrainerGeneratorFactory.cs (1)
30
case TrainerName.
LightGbmRegression
: