28 references to MulticlassClassificationTrainer
Microsoft.ML.AutoML (18)
API\MulticlassClassificationExperiment.cs (10)
35/// The default value is a collection auto-populated with all possible trainers (all values of <see cref="MulticlassClassificationTrainer" />). 37public ICollection<MulticlassClassificationTrainer> Trainers { get; } 45Trainers = Enum.GetValues(typeof(MulticlassClassificationTrainer)).OfType<MulticlassClassificationTrainer>().ToList(); 316var useSdcaMaximumEntrophy = Settings.Trainers.Contains(MulticlassClassificationTrainer.SdcaMaximumEntropy); 317var uselbfgsLR = Settings.Trainers.Contains(MulticlassClassificationTrainer.LbfgsLogisticRegressionOva); 318var uselbfgsME = Settings.Trainers.Contains(MulticlassClassificationTrainer.LbfgsMaximumEntropy); 319var useLgbm = Settings.Trainers.Contains(MulticlassClassificationTrainer.LightGbm); 320var useFastForest = Settings.Trainers.Contains(MulticlassClassificationTrainer.FastForestOva); 321var useFastTree = Settings.Trainers.Contains(MulticlassClassificationTrainer.FastTreeOva);
TrainerExtensions\TrainerExtensionUtil.cs (8)
321public static TrainerName GetTrainerName(MulticlassClassificationTrainer multiTrainer) 325case MulticlassClassificationTrainer.FastForestOva: 327case MulticlassClassificationTrainer.FastTreeOva: 329case MulticlassClassificationTrainer.LightGbm: 331case MulticlassClassificationTrainer.LbfgsMaximumEntropy: 333case MulticlassClassificationTrainer.LbfgsLogisticRegressionOva: 335case MulticlassClassificationTrainer.SdcaMaximumEntropy: 396public static IEnumerable<TrainerName> GetTrainerNames(IEnumerable<MulticlassClassificationTrainer> multiTrainers)
Microsoft.ML.AutoML.Tests (10)
AutoFitTests.cs (8)
280settings.Trainers.Remove(MulticlassClassificationTrainer.LightGbm); 281settings.Trainers.Remove(MulticlassClassificationTrainer.SdcaMaximumEntropy); 282settings.Trainers.Remove(MulticlassClassificationTrainer.LbfgsMaximumEntropy); 283settings.Trainers.Remove(MulticlassClassificationTrainer.LbfgsLogisticRegressionOva); 312settings.Trainers.Remove(MulticlassClassificationTrainer.LightGbm); 313settings.Trainers.Remove(MulticlassClassificationTrainer.SdcaMaximumEntropy); 314settings.Trainers.Remove(MulticlassClassificationTrainer.LbfgsMaximumEntropy); 315settings.Trainers.Remove(MulticlassClassificationTrainer.LbfgsLogisticRegressionOva);
TrainerExtensionsTests.cs (2)
365var publicNames = Enum.GetValues(typeof(MulticlassClassificationTrainer)).Cast<MulticlassClassificationTrainer>();