32 references to BinaryClassificationTrainer
Microsoft.ML.AutoML (16)
API\BinaryClassificationExperiment.cs (9)
36/// <value>The default value is a collection auto-populated with all possible trainers (all values of <see cref="BinaryClassificationTrainer" />).</value> 37public ICollection<BinaryClassificationTrainer> Trainers { get; } 50Trainers = Enum.GetValues(typeof(BinaryClassificationTrainer)).OfType<BinaryClassificationTrainer>().ToList(); 326var useSdca = Settings.Trainers.Contains(BinaryClassificationTrainer.SdcaLogisticRegression); 327var uselbfgs = Settings.Trainers.Contains(BinaryClassificationTrainer.LbfgsLogisticRegression); 328var useLgbm = Settings.Trainers.Contains(BinaryClassificationTrainer.LightGbm); 329var useFastForest = Settings.Trainers.Contains(BinaryClassificationTrainer.FastForest); 330var useFastTree = Settings.Trainers.Contains(BinaryClassificationTrainer.FastTree);
TrainerExtensions\TrainerExtensionUtil.cs (7)
301public static TrainerName GetTrainerName(BinaryClassificationTrainer binaryTrainer) 305case BinaryClassificationTrainer.FastForest: 307case BinaryClassificationTrainer.FastTree: 309case BinaryClassificationTrainer.LightGbm: 311case BinaryClassificationTrainer.LbfgsLogisticRegression: 313case BinaryClassificationTrainer.SdcaLogisticRegression: 391public static IEnumerable<TrainerName> GetTrainerNames(IEnumerable<BinaryClassificationTrainer> binaryTrainers)
Microsoft.ML.AutoML.Tests (16)
AutoFitTests.cs (13)
54settings.Trainers.Remove(BinaryClassificationTrainer.LightGbm); 55settings.Trainers.Remove(BinaryClassificationTrainer.SdcaLogisticRegression); 56settings.Trainers.Remove(BinaryClassificationTrainer.LbfgsLogisticRegression); 81settings.Trainers.Remove(BinaryClassificationTrainer.LightGbm); 82settings.Trainers.Remove(BinaryClassificationTrainer.SdcaLogisticRegression); 83settings.Trainers.Remove(BinaryClassificationTrainer.LbfgsLogisticRegression); 108settings.Trainers.Remove(BinaryClassificationTrainer.LightGbm); 109settings.Trainers.Remove(BinaryClassificationTrainer.SdcaLogisticRegression); 110settings.Trainers.Remove(BinaryClassificationTrainer.LbfgsLogisticRegression); 134settings.Trainers.Remove(BinaryClassificationTrainer.LightGbm); 135settings.Trainers.Remove(BinaryClassificationTrainer.SdcaLogisticRegression); 136settings.Trainers.Remove(BinaryClassificationTrainer.LbfgsLogisticRegression); 598settings.Trainers.Remove(BinaryClassificationTrainer.FastForest);
TrainerExtensionsTests.cs (3)
357var publicNames = Enum.GetValues(typeof(BinaryClassificationTrainer)).Cast<BinaryClassificationTrainer>(); 397var internalNames = TrainerExtensionUtil.GetTrainerNames(null as IEnumerable<BinaryClassificationTrainer>);