@@ -21,7 +21,7 @@ namespace Microsoft.ML.LightGBM
21
21
{
22
22
23
23
/// <include file='doc.xml' path='doc/members/member[@name="LightGBM"]/*' />
24
- public sealed class LightGbmMulticlassTrainer : LightGbmTrainerBase < VBuffer < float > , MulticlassPredictionTransformer < OvaModelParameters > , OvaModelParameters >
24
+ public sealed class LightGbmMulticlassTrainer : LightGbmTrainerBase < VBuffer < float > , MulticlassPredictionTransformer < OneVersusAllModelParameters > , OneVersusAllModelParameters >
25
25
{
26
26
internal const string Summary = "LightGBM Multi Class Classifier" ;
27
27
internal const string LoadNameValue = "LightGBMMulticlass" ;
@@ -80,7 +80,7 @@ private LightGbmBinaryModelParameters CreateBinaryPredictor(int classID, string
80
80
return new LightGbmBinaryModelParameters ( Host , GetBinaryEnsemble ( classID ) , FeatureCount , innerArgs ) ;
81
81
}
82
82
83
- private protected override OvaModelParameters CreatePredictor ( )
83
+ private protected override OneVersusAllModelParameters CreatePredictor ( )
84
84
{
85
85
Host . Check ( TrainedEnsemble != null , "The predictor cannot be created before training is complete." ) ;
86
86
@@ -97,9 +97,9 @@ private protected override OvaModelParameters CreatePredictor()
97
97
}
98
98
string obj = ( string ) GetGbmParameters ( ) [ "objective" ] ;
99
99
if ( obj == "multiclass" )
100
- return OvaModelParameters . Create ( Host , OvaModelParameters . OutputFormula . Softmax , predictors ) ;
100
+ return OneVersusAllModelParameters . Create ( Host , OneVersusAllModelParameters . OutputFormula . Softmax , predictors ) ;
101
101
else
102
- return OvaModelParameters . Create ( Host , predictors ) ;
102
+ return OneVersusAllModelParameters . Create ( Host , predictors ) ;
103
103
}
104
104
105
105
private protected override void CheckDataValid ( IChannel ch , RoleMappedData data )
@@ -218,14 +218,14 @@ private protected override SchemaShape.Column[] GetOutputColumnsCore(SchemaShape
218
218
} ;
219
219
}
220
220
221
- private protected override MulticlassPredictionTransformer < OvaModelParameters > MakeTransformer ( OvaModelParameters model , DataViewSchema trainSchema )
222
- => new MulticlassPredictionTransformer < OvaModelParameters > ( Host , model , trainSchema , FeatureColumn . Name , LabelColumn . Name ) ;
221
+ private protected override MulticlassPredictionTransformer < OneVersusAllModelParameters > MakeTransformer ( OneVersusAllModelParameters model , DataViewSchema trainSchema )
222
+ => new MulticlassPredictionTransformer < OneVersusAllModelParameters > ( Host , model , trainSchema , FeatureColumn . Name , LabelColumn . Name ) ;
223
223
224
224
/// <summary>
225
225
/// Trains a <see cref="LightGbmMulticlassTrainer"/> using both training and validation data, returns
226
- /// a <see cref="MulticlassPredictionTransformer{OvaModelParameters }"/>.
226
+ /// a <see cref="MulticlassPredictionTransformer{OneVsAllModelParameters }"/>.
227
227
/// </summary>
228
- public MulticlassPredictionTransformer < OvaModelParameters > Fit ( IDataView trainData , IDataView validationData )
228
+ public MulticlassPredictionTransformer < OneVersusAllModelParameters > Fit ( IDataView trainData , IDataView validationData )
229
229
=> TrainTransformer ( trainData , validationData ) ;
230
230
}
231
231
0 commit comments