public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
return(TrainerExtensionUtil.BuildLightGbmPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams,
columnInfo.LabelColumnName, columnInfo.ExampleWeightColumnName));
}
public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable <SweepableParam> sweepParams,
ColumnInformation columnInfo, IDataView validationSet)
{
LightGbmMulticlassTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions <LightGbmMulticlassTrainer.Options, VBuffer <float>, MulticlassPredictionTransformer <OneVersusAllModelParameters>, OneVersusAllModelParameters>(sweepParams, columnInfo);
return(mlContext.MulticlassClassification.Trainers.LightGbm(options));
}
public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable <SweepableParam> sweepParams,
ColumnInformation columnInfo, IDataView validationSet)
{
LightGbmRegressionTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions <LightGbmRegressionTrainer.Options, float, RegressionPredictionTransformer <LightGbmRegressionModelParameters>, LightGbmRegressionModelParameters>(sweepParams, columnInfo);
return(mlContext.Regression.Trainers.LightGbm(options));
}
public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
return(TrainerExtensionUtil.BuildOvaPipelineNode(this, _binaryLearnerCatalogItem, sweepParams, columnInfo));
}
public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable <SweepableParam> sweepParams,
ColumnInformation columnInfo)
{
LightGbmBinaryTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions <LightGbmBinaryTrainer.Options, float, BinaryPredictionTransformer <CalibratedModelParametersBase <LightGbmBinaryModelParameters, PlattCalibrator> >, CalibratedModelParametersBase <LightGbmBinaryModelParameters, PlattCalibrator> >(sweepParams, columnInfo);
return(mlContext.BinaryClassification.Trainers.LightGbm(options));
}
