Exemple #1
0
        public static CommonOutputs.BinaryClassificationOutput TrainBinary(IHostEnvironment env, FastTreeBinaryTrainer.Options input)
        {
            Contracts.CheckValue(env, nameof(env));
            var host = env.Register("TrainFastTree");

            host.CheckValue(input, nameof(input));
            EntryPointUtils.CheckInputArgs(host, input);

            return(TrainerEntryPointsUtils.Train <FastTreeBinaryTrainer.Options, CommonOutputs.BinaryClassificationOutput>(host, input,
                                                                                                                           () => new FastTreeBinaryTrainer(host, input),
                                                                                                                           () => TrainerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.LabelColumnName),
                                                                                                                           () => TrainerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.ExampleWeightColumnName),
                                                                                                                           () => TrainerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.RowGroupColumnName)));
        }
 internal FastTreeBinaryFeaturizationEstimator(IHostEnvironment env, Options options)
     : base(env, options)
 {
     _trainerOptions = options.TrainerOptions;
 }