1 write to LightGbmTrainerOptions
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
354LightGbmTrainerOptions = options;
50 references to LightGbmTrainerOptions
Microsoft.ML.LightGbm (50)
LightGbmBinaryTrainer.cs (2)
258var cali = new PlattCalibrator(Host, -LightGbmTrainerOptions.Sigmoid, 0); 268if (LightGbmTrainerOptions.LightGbmModel == null)
LightGbmMulticlassTrainer.cs (14)
223var cali = new PlattCalibrator(Host, -LightGbmTrainerOptions.Sigmoid, 0); 238if (LightGbmTrainerOptions.LightGbmModel == null) 321int minimumExampleCountPerLeaf = LightGbmTrainerOptions.MinimumExampleCountPerLeaf ?? DefaultMinDataPerLeaf(numRow, numberOfLeaves, _numberOfClassesIncludingNan); 325if (!LightGbmTrainerOptions.LearningRate.HasValue) 326ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.LearningRate) + " = " + GbmOptions["learning_rate"]); 327if (!LightGbmTrainerOptions.NumberOfLeaves.HasValue) 328ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.NumberOfLeaves) + " = " + numberOfLeaves); 329if (!LightGbmTrainerOptions.MinimumExampleCountPerLeaf.HasValue) 330ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.MinimumExampleCountPerLeaf) + " = " + minimumExampleCountPerLeaf); 342if (LightGbmTrainerOptions.UseSoftmax.HasValue) 343useSoftmax = LightGbmTrainerOptions.UseSoftmax.Value; 349ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.UseSoftmax) + " = " + useSoftmax); 361if (LightGbmTrainerOptions.LightGbmModel == null) 367var 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)
245if (LightGbmTrainerOptions.LightGbmModel == null)
LightGbmRegressionTrainer.cs (1)
230if (LightGbmTrainerOptions.LightGbmModel == null)
LightGbmTrainerBase.cs (32)
355ParallelTraining = LightGbmTrainerOptions.ParallelTrainer != null ? LightGbmTrainerOptions.ParallelTrainer.CreateComponent(Host) : new SingleTrainer(); 356GbmOptions = LightGbmTrainerOptions.ToDictionary(Host); 370if (LightGbmTrainerOptions.LightGbmModel != null) 372LightGbmTrainerOptions.LightGbmModel.Position = 0; 375StreamReader reader = new StreamReader(LightGbmTrainerOptions.LightGbmModel); 393Type optionsType = LightGbmTrainerOptions.GetType(); 398var lightGbmName = LightGbmTrainerOptions.GetOptionName(field.Name); 405field.SetValue(LightGbmTrainerOptions, Double.Parse(modelParameters[lightGbmName])); 409field.SetValue(LightGbmTrainerOptions, int.Parse(modelParameters[lightGbmName])); 413field.SetValue(LightGbmTrainerOptions, float.Parse(modelParameters[lightGbmName])); 420field.SetValue(LightGbmTrainerOptions, true); 422field.SetValue(LightGbmTrainerOptions, false); 425field.SetValue(LightGbmTrainerOptions, Convert.ChangeType(modelParameters[lightGbmName], field.FieldType)); 501if (LightGbmTrainerOptions.LightGbmModel == null) 507double learningRate = LightGbmTrainerOptions.LearningRate ?? DefaultLearningRate(numRow, hasCategorical, totalCats); 508int numberOfLeaves = LightGbmTrainerOptions.NumberOfLeaves ?? DefaultNumLeaves(numRow, hasCategorical, totalCats); 509int minimumExampleCountPerLeaf = LightGbmTrainerOptions.MinimumExampleCountPerLeaf ?? DefaultMinDataPerLeaf(numRow, numberOfLeaves, 1); 515if (!LightGbmTrainerOptions.LearningRate.HasValue) 516ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.LearningRate) + " = " + learningRate); 517if (!LightGbmTrainerOptions.NumberOfLeaves.HasValue) 518ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.NumberOfLeaves) + " = " + numberOfLeaves); 519if (!LightGbmTrainerOptions.MinimumExampleCountPerLeaf.HasValue) 520ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.MinimumExampleCountPerLeaf) + " = " + minimumExampleCountPerLeaf); 616bool useCat = LightGbmTrainerOptions.UseCategoricalSplit ?? numRow > useCatThreshold; 617if (!LightGbmTrainerOptions.UseCategoricalSplit.HasValue) 618ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.UseCategoricalSplit) + " = " + useCat); 668LoadDataset(ch, factory, dtrain, numRow, LightGbmTrainerOptions.BatchSize, catMetaData); 690LoadDataset(ch, factory, dvalid, numRow, LightGbmTrainerOptions.BatchSize, catMetaData); 712dvalid: dvalid, numIteration: LightGbmTrainerOptions.NumberOfIterations, 713verboseEval: LightGbmTrainerOptions.Verbose, earlyStoppingRound: LightGbmTrainerOptions.EarlyStoppingRound))