31 implementations of ITrainerExtension
Microsoft.ML.AutoML (31)
TrainerExtensions\BinaryTrainerExtensions.cs (9)
17internal class AveragedPerceptronBinaryExtension : ITrainerExtension 64internal class FastForestBinaryExtension : ITrainerExtension 86internal class FastTreeBinaryExtension : ITrainerExtension 108internal class LightGbmBinaryExtension : ITrainerExtension 129internal class LinearSvmBinaryExtension : ITrainerExtension 150internal class SdcaLogisticRegressionBinaryExtension : ITrainerExtension 171internal class LbfgsLogisticRegressionBinaryExtension : ITrainerExtension 193internal class SgdCalibratedBinaryExtension : ITrainerExtension 215internal class SymbolicSgdLogisticRegressionBinaryExtension : ITrainerExtension
TrainerExtensions\MultiTrainerExtensions.cs (11)
16internal class AveragedPerceptronOvaExtension : ITrainerExtension 38internal class FastForestOvaExtension : ITrainerExtension 60internal class LightGbmMultiExtension : ITrainerExtension 81internal class LinearSvmOvaExtension : ITrainerExtension 103internal class SdcaMaximumEntropyMultiExtension : ITrainerExtension 124internal class LbfgsLogisticRegressionOvaExtension : ITrainerExtension 146internal class SgdCalibratedOvaExtension : ITrainerExtension 168internal class SymbolicSgdLogisticRegressionOvaExtension : ITrainerExtension 190internal class FastTreeOvaExtension : ITrainerExtension 212internal class LbfgsMaximumEntropyMultiExtension : ITrainerExtension 234internal class ImageClassificationExtension : ITrainerExtension
TrainerExtensions\RankingTrainerExtensions.cs (2)
14internal class LightGbmRankingExtension : ITrainerExtension 37internal class FastTreeRankingExtension : ITrainerExtension
TrainerExtensions\RecommendationTrainerExtensions.cs (1)
12internal class MatrixFactorizationExtension : ITrainerExtension
TrainerExtensions\RegressionTrainerExtensions.cs (8)
15internal class FastForestRegressionExtension : ITrainerExtension 37internal class FastTreeRegressionExtension : ITrainerExtension 59internal class FastTreeTweedieRegressionExtension : ITrainerExtension 81internal class LightGbmRegressionExtension : ITrainerExtension 102internal class OnlineGradientDescentRegressionExtension : ITrainerExtension 123internal class OlsRegressionExtension : ITrainerExtension 145internal class LbfgsPoissonRegressionExtension : ITrainerExtension 167internal class SdcaRegressionExtension : ITrainerExtension
31 references to ITrainerExtension
Microsoft.ML.AutoML (29)
Experiment\RecipeInference.cs (1)
21foreach (var trainerExtension in trainerExtensions)
Experiment\SuggestedPipeline.cs (1)
87var trainerExtension = TrainerExtensionCatalog.GetTrainerExtension(trainerName);
Experiment\SuggestedTrainer.cs (2)
18private readonly ITrainerExtension _trainerExtension; 21internal SuggestedTrainer(MLContext mlContext, ITrainerExtension trainerExtension,
TrainerExtensions\MultiTrainerExtensions.cs (7)
18private static readonly ITrainerExtension _binaryLearnerCatalogItem = new AveragedPerceptronBinaryExtension(); 40private static readonly ITrainerExtension _binaryLearnerCatalogItem = new FastForestBinaryExtension(); 83private static readonly ITrainerExtension _binaryLearnerCatalogItem = new LinearSvmBinaryExtension(); 126private static readonly ITrainerExtension _binaryLearnerCatalogItem = new LbfgsLogisticRegressionBinaryExtension(); 148private static readonly ITrainerExtension _binaryLearnerCatalogItem = new SgdCalibratedBinaryExtension(); 170private static readonly ITrainerExtension _binaryLearnerCatalogItem = new SymbolicSgdLogisticRegressionBinaryExtension(); 192private static readonly ITrainerExtension _binaryLearnerCatalogItem = new FastTreeBinaryExtension();
TrainerExtensions\TrainerExtensionCatalog.cs (16)
52public static TrainerName GetTrainerName(ITrainerExtension trainerExtension) 57public static ITrainerExtension GetTrainerExtension(TrainerName trainerName) 60return (ITrainerExtension)Activator.CreateInstance(trainerExtensionType); 63public static IEnumerable<ITrainerExtension> GetTrainers(TaskKind task, 66IEnumerable<ITrainerExtension> trainers; 80return new List<ITrainerExtension>() { new ImageClassificationExtension() }; 113private static IEnumerable<ITrainerExtension> GetBinaryLearners() 115return new ITrainerExtension[] 129private static IEnumerable<ITrainerExtension> GetMultiLearners() 131return new ITrainerExtension[] 146private static IEnumerable<ITrainerExtension> GetRegressionLearners() 148return new ITrainerExtension[] 161private static IEnumerable<ITrainerExtension> GetRecommendationLearners() 163return new ITrainerExtension[] 169private static IEnumerable<ITrainerExtension> GetRankingLearners() 171return new ITrainerExtension[]
TrainerExtensions\TrainerExtensionUtil.cs (2)
100public static PipelineNode BuildOvaPipelineNode(ITrainerExtension multiExtension, ITrainerExtension binaryExtension,
Microsoft.ML.AutoML.Tests (2)
TrainerExtensionsTests.cs (2)
31var extension = TrainerExtensionCatalog.GetTrainerExtension(trainerName); 56var extension = TrainerExtensionCatalog.GetTrainerExtension(TrainerName.ImageClassification);