diff --git a/src/Microsoft.ML.LightGbm/LightGbmCatalog.cs b/src/Microsoft.ML.LightGbm/LightGbmCatalog.cs index 40d5532cfb..5cf883cc25 100644 --- a/src/Microsoft.ML.LightGbm/LightGbmCatalog.cs +++ b/src/Microsoft.ML.LightGbm/LightGbmCatalog.cs @@ -173,7 +173,7 @@ public static LightGbmRankingTrainer LightGbm(this RankingCatalog.RankingTrainer } /// - /// Predict a target using a gradient boosting decision tree multiclass classification model trained with the . + /// Create , which predicts a target using a gradient boosting decision tree multiclass classification model. /// /// The . /// The name of the label column. @@ -205,7 +205,7 @@ public static LightGbmMulticlassTrainer LightGbm(this MulticlassClassificationCa } /// - /// Predict a target using a gradient boosting decision tree multiclass classification model trained with the and advanced options. + /// Create with advanced options, which predicts a target using a gradient boosting decision tree multiclass classification model. /// /// The . /// Trainer options. diff --git a/src/Microsoft.ML.LightGbm/LightGbmMulticlassTrainer.cs b/src/Microsoft.ML.LightGbm/LightGbmMulticlassTrainer.cs index 1475f9a4a5..5c76eec2a1 100644 --- a/src/Microsoft.ML.LightGbm/LightGbmMulticlassTrainer.cs +++ b/src/Microsoft.ML.LightGbm/LightGbmMulticlassTrainer.cs @@ -24,7 +24,28 @@ namespace Microsoft.ML.Trainers.LightGbm /// /// The for training a boosted decision tree multi-class classification model using LightGBM. /// - /// + /// + /// + /// + /// + /// + /// + /// public sealed class LightGbmMulticlassTrainer : LightGbmTrainerBase, MulticlassPredictionTransformer, @@ -40,6 +61,10 @@ public sealed class LightGbmMulticlassTrainer : LightGbmTrainerBase PredictionKind.MulticlassClassification; + /// + /// Options for the as used in + /// [LightGbm(Options)](xref:Microsoft.ML.LightGbmExtensions.LightGbm(Microsoft.ML.MulticlassClassificationCatalog.MulticlassClassificationTrainers,Microsoft.ML.Trainers.LightGbm.LightGbmMulticlassTrainer.Options)). + /// public sealed class Options : OptionsBase { public enum EvaluateMetricType