namespace Microsoft.ML.Trainers.FastTree
{
    public sealed class FastTreeTweedieTrainer : BoostingFastTreeTrainerBase<Options, RegressionPredictionTransformer<FastTreeTweedieModelParameters>, FastTreeTweedieModelParameters>
    {
        [Component(Desc = "Trains gradient boosted decision trees to fit target values using a Tweedie loss function. This learner is a generalization of Poisson, compound Poisson, and gamma regression.", FriendlyName = "FastTree (Boosted Trees) Tweedie Regression", Name = "FastTreeTweedieRegression")]
        public sealed class Options : BoostedTreeOptions, IComponentFactory, IComponentFactory<ITrainer>
        {
            public EarlyStoppingMetric EarlyStoppingMetric { get; set; }
        }
    }
}