29 references to RegressionTrainer
Microsoft.ML.AutoML (18)
API\RegressionExperiment.cs (10)
35/// The default value is a collection auto-populated with all possible trainers (all values of <see cref="RegressionTrainer" />). 37public ICollection<RegressionTrainer> Trainers { get; } 45Trainers = Enum.GetValues(typeof(RegressionTrainer)).OfType<RegressionTrainer>().ToList(); 299var useSdca = Settings.Trainers.Contains(RegressionTrainer.StochasticDualCoordinateAscent); 300var uselbfgs = Settings.Trainers.Contains(RegressionTrainer.LbfgsPoissonRegression); 301var useLgbm = Settings.Trainers.Contains(RegressionTrainer.LightGbm); 302var useFastForest = Settings.Trainers.Contains(RegressionTrainer.FastForest); 303var useFastTree = Settings.Trainers.Contains(RegressionTrainer.FastTree) || Settings.Trainers.Contains(RegressionTrainer.FastTreeTweedie);
TrainerExtensions\TrainerExtensionUtil.cs (8)
343public static TrainerName GetTrainerName(RegressionTrainer regressionTrainer) 347case RegressionTrainer.FastForest: 349case RegressionTrainer.FastTree: 351case RegressionTrainer.FastTreeTweedie: 353case RegressionTrainer.LightGbm: 355case RegressionTrainer.LbfgsPoissonRegression: 357case RegressionTrainer.StochasticDualCoordinateAscent: 401public static IEnumerable<TrainerName> GetTrainerNames(IEnumerable<RegressionTrainer> regressionTrainers)
Microsoft.ML.AutoML.Tests (11)
AutoFitTests.cs (9)
167settings.Trainers.Remove(RegressionTrainer.LightGbm); 168settings.Trainers.Remove(RegressionTrainer.StochasticDualCoordinateAscent); 169settings.Trainers.Remove(RegressionTrainer.LbfgsPoissonRegression); 197settings.Trainers.Remove(RegressionTrainer.LightGbm); 198settings.Trainers.Remove(RegressionTrainer.StochasticDualCoordinateAscent); 199settings.Trainers.Remove(RegressionTrainer.LbfgsPoissonRegression); 227settings.Trainers.Remove(RegressionTrainer.LightGbm); 228settings.Trainers.Remove(RegressionTrainer.StochasticDualCoordinateAscent); 229settings.Trainers.Remove(RegressionTrainer.LbfgsPoissonRegression);
TrainerExtensionsTests.cs (2)
373var publicNames = Enum.GetValues(typeof(RegressionTrainer)).Cast<RegressionTrainer>();