1 write to LightGbmTrainerOptions
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
378
LightGbmTrainerOptions
= options;
50 references to LightGbmTrainerOptions
Microsoft.ML.LightGbm (50)
LightGbmBinaryTrainer.cs (2)
258
var cali = new PlattCalibrator(Host, -
LightGbmTrainerOptions
.Sigmoid, 0);
268
if (
LightGbmTrainerOptions
.LightGbmModel == null)
LightGbmMulticlassTrainer.cs (14)
223
var cali = new PlattCalibrator(Host, -
LightGbmTrainerOptions
.Sigmoid, 0);
238
if (
LightGbmTrainerOptions
.LightGbmModel == null)
321
int minimumExampleCountPerLeaf =
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf ?? DefaultMinDataPerLeaf(numRow, numberOfLeaves, _numberOfClassesIncludingNan);
325
if (!
LightGbmTrainerOptions
.LearningRate.HasValue)
326
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.LearningRate) + " = " + GbmOptions["learning_rate"]);
327
if (!
LightGbmTrainerOptions
.NumberOfLeaves.HasValue)
328
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.NumberOfLeaves) + " = " + numberOfLeaves);
329
if (!
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf.HasValue)
330
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf) + " = " + minimumExampleCountPerLeaf);
342
if (
LightGbmTrainerOptions
.UseSoftmax.HasValue)
343
useSoftmax =
LightGbmTrainerOptions
.UseSoftmax.Value;
349
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.UseSoftmax) + " = " + useSoftmax);
361
if (
LightGbmTrainerOptions
.LightGbmModel == null)
367
var metadata =
LightGbmTrainerOptions
.LightGbmModel == null ? new SchemaShape(labelCol.Annotations.Where(x => x.Name == AnnotationUtils.Kinds.KeyValues).Concat(AnnotationUtils.GetTrainerOutputAnnotation())) : new SchemaShape(AnnotationUtils.GetTrainerOutputAnnotation());
LightGbmRankingTrainer.cs (1)
245
if (
LightGbmTrainerOptions
.LightGbmModel == null)
LightGbmRegressionTrainer.cs (1)
230
if (
LightGbmTrainerOptions
.LightGbmModel == null)
LightGbmTrainerBase.cs (32)
379
ParallelTraining =
LightGbmTrainerOptions
.ParallelTrainer != null ?
LightGbmTrainerOptions
.ParallelTrainer.CreateComponent(Host) : new SingleTrainer();
380
GbmOptions =
LightGbmTrainerOptions
.ToDictionary(Host);
394
if (
LightGbmTrainerOptions
.LightGbmModel != null)
396
LightGbmTrainerOptions
.LightGbmModel.Position = 0;
399
StreamReader reader = new StreamReader(
LightGbmTrainerOptions
.LightGbmModel);
417
Type optionsType =
LightGbmTrainerOptions
.GetType();
422
var lightGbmName =
LightGbmTrainerOptions
.GetOptionName(field.Name);
429
field.SetValue(
LightGbmTrainerOptions
, Double.Parse(modelParameters[lightGbmName]));
433
field.SetValue(
LightGbmTrainerOptions
, int.Parse(modelParameters[lightGbmName]));
437
field.SetValue(
LightGbmTrainerOptions
, float.Parse(modelParameters[lightGbmName]));
444
field.SetValue(
LightGbmTrainerOptions
, true);
446
field.SetValue(
LightGbmTrainerOptions
, false);
449
field.SetValue(
LightGbmTrainerOptions
, Convert.ChangeType(modelParameters[lightGbmName], field.FieldType));
525
if (
LightGbmTrainerOptions
.LightGbmModel == null)
531
double learningRate =
LightGbmTrainerOptions
.LearningRate ?? DefaultLearningRate(numRow, hasCategorical, totalCats);
532
int numberOfLeaves =
LightGbmTrainerOptions
.NumberOfLeaves ?? DefaultNumLeaves(numRow, hasCategorical, totalCats);
533
int minimumExampleCountPerLeaf =
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf ?? DefaultMinDataPerLeaf(numRow, numberOfLeaves, 1);
539
if (!
LightGbmTrainerOptions
.LearningRate.HasValue)
540
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.LearningRate) + " = " + learningRate);
541
if (!
LightGbmTrainerOptions
.NumberOfLeaves.HasValue)
542
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.NumberOfLeaves) + " = " + numberOfLeaves);
543
if (!
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf.HasValue)
544
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.MinimumExampleCountPerLeaf) + " = " + minimumExampleCountPerLeaf);
640
bool useCat =
LightGbmTrainerOptions
.UseCategoricalSplit ?? numRow > useCatThreshold;
641
if (!
LightGbmTrainerOptions
.UseCategoricalSplit.HasValue)
642
ch.Info("Auto-tuning parameters: " + nameof(
LightGbmTrainerOptions
.UseCategoricalSplit) + " = " + useCat);
692
LoadDataset(ch, factory, dtrain, numRow,
LightGbmTrainerOptions
.BatchSize, catMetaData);
714
LoadDataset(ch, factory, dvalid, numRow,
LightGbmTrainerOptions
.BatchSize, catMetaData);
736
dvalid: dvalid, numIteration:
LightGbmTrainerOptions
.NumberOfIterations,
737
verboseEval:
LightGbmTrainerOptions
.Verbose, earlyStoppingRound:
LightGbmTrainerOptions
.EarlyStoppingRound))