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)
1056UseCategoricalSplit = true 1091UseCategoricalSplit = true 1119UseCategoricalSplit = true
3 references to UseCategoricalSplit
Microsoft.ML.LightGbm (3)
LightGbmTrainerBase.cs (3)
640bool useCat = LightGbmTrainerOptions.UseCategoricalSplit ?? numRow > useCatThreshold; 641if (!LightGbmTrainerOptions.UseCategoricalSplit.HasValue) 642ch.Info("Auto-tuning parameters: " + nameof(LightGbmTrainerOptions.UseCategoricalSplit) + " = " + useCat);