public ObjectiveImpl(Dataset trainData, GamRegressionTrainer.Options options) : base( trainData, options.LearningRate, 0, options.MaximumTreeOutput, options.GetDerivativesSampleRate, false, options.Seed) { _labels = GetDatasetRegressionLabels(trainData); }
public static CommonOutputs.RegressionOutput TrainRegression(IHostEnvironment env, GamRegressionTrainer.Options input) { Contracts.CheckValue(env, nameof(env)); var host = env.Register("TrainGAM"); host.CheckValue(input, nameof(input)); EntryPointUtils.CheckInputArgs(host, input); return(TrainerEntryPointsUtils.Train <GamRegressionTrainer.Options, CommonOutputs.RegressionOutput>(host, input, () => new GamRegressionTrainer(host, input), () => TrainerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.LabelColumnName), () => TrainerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.ExampleWeightColumnName))); }