public static CommonOutputs.RegressionOutput TrainRegression(IHostEnvironment env, FastTreeRegressionTrainer.Options input)
{
Contracts.CheckValue(env, nameof(env));
var host = env.Register("TrainFastTree");
host.CheckValue(input, nameof(input));
EntryPointUtils.CheckInputArgs(host, input);
return(TrainerEntryPointsUtils.Train <FastTreeRegressionTrainer.Options, CommonOutputs.RegressionOutput>(host, input,
() => new FastTreeRegressionTrainer(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 FastTreeRegressionFeaturizationEstimator(IHostEnvironment env, Options options)
: base(env, options)
{
_trainerOptions = options.TrainerOptions;
}
