namespace Microsoft.ML.Trainers.LightGbm
{
    public abstract class LightGbmTrainerBase<TOptions, TOutput, TTransformer, TModel> : TrainerEstimatorBaseWithGroupId<TTransformer, TModel>
        where TOptions, new(), OptionsBase
        where TTransformer, ISingleFeaturePredictionTransformer<TModel>
        where TModel, class
    {
        public class OptionsBase : TrainerInputBaseWithGroupId
        {
            [SweepableDiscreteParam("UseCat", new[] { true, false }, false)]
            public bool? UseCategoricalSplit;
        }
    }
}