private protected LightGbmTrainerBase(IHostEnvironment env, string name, LightGbmArguments args, SchemaShape.Column label)
: base(Contracts.CheckRef(env, nameof(env)).Register(name), TrainerUtils.MakeR4VecFeature(args.FeatureColumn), label, TrainerUtils.MakeR4ScalarWeightColumn(args.WeightColumn))
{
Host.CheckValue(args, nameof(args));
Args = args;
InitParallelTraining();
}
private protected LightGbmTrainerBase(IHostEnvironment env,
string name,
SchemaShape.Column label,
string featureColumn,
string weightColumn,
string groupIdColumn,
int?numLeaves,
int?minDataPerLeaf,
double?learningRate,
int numBoostRound,
Action <LightGbmArguments> advancedSettings)
: base(Contracts.CheckRef(env, nameof(env)).Register(name), TrainerUtils.MakeR4VecFeature(featureColumn), label, TrainerUtils.MakeR4ScalarWeightColumn(weightColumn), TrainerUtils.MakeU4ScalarColumn(groupIdColumn))
{
Args = new LightGbmArguments();
Args.NumLeaves = numLeaves;
Args.MinDataPerLeaf = minDataPerLeaf;
Args.LearningRate = learningRate;
Args.NumBoostRound = numBoostRound;
//apply the advanced args, if the user supplied any
advancedSettings?.Invoke(Args);
Args.LabelColumn = label.Name;
Args.FeatureColumn = featureColumn;
if (weightColumn != null)
{
Args.WeightColumn = Optional <string> .Explicit(weightColumn);
}
if (groupIdColumn != null)
{
Args.GroupIdColumn = Optional <string> .Explicit(groupIdColumn);
}
InitParallelTraining();
}
internal LightGbmBinaryTrainer(IHostEnvironment env, LightGbmArguments args)
: base(env, LoadNameValue, args, TrainerUtils.MakeBoolScalarLabel(args.LabelColumn))
{
}
internal LightGbmMulticlassTrainer(IHostEnvironment env, LightGbmArguments args)
: base(env, LoadNameValue, args, TrainerUtils.MakeBoolScalarLabel(args.LabelColumn))
{
_numClass = -1;
}
