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