namespace Microsoft.ML.Trainers.FastTree
{
    public sealed class FastTreeBinaryTrainer : BoostingFastTreeTrainerBase<Options, BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>, CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>
    {
        [Component(Desc = "Uses a logit-boost boosted tree learner to perform binary classification.", FriendlyName = "FastTree (Boosted Trees) Classification", Name = "FastTreeBinaryClassification")]
        public sealed class Options : BoostedTreeOptions, IComponentFactory, IComponentFactory<ITrainer>
        {
            public bool UnbalancedSets;
        }
    }
}