4 writes to UseCategoricalSplit
Microsoft.ML.Tests (4)
FeatureContributionTests.cs (1)
59TestFeatureContribution(ML.Regression.Trainers.LightGbm(new LightGbmRegressionTrainer.Options() { UseCategoricalSplit = true }), GetOneHotEncodedData(numberOfInstances: 100), "LightGbmRegressionWithCategoricalSplit");
TrainerEstimators\TreeEstimators.cs (3)
1054UseCategoricalSplit = true 1089UseCategoricalSplit = true 1117UseCategoricalSplit = true
3 references to UseCategoricalSplit
Microsoft.ML.LightGbm (3)
LightGbmTrainerBase.cs (3)
616bool useCat = LightGbmTrainerOptions.UseCategoricalSplit ?? numRow > useCatThreshold; 617if (!LightGbmTrainerOptions.UseCategoricalSplit.HasValue) 618ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.UseCategoricalSplit) + " = " + useCat);