@@ -165,8 +165,13 @@ public sealed class Arguments : ArgumentsBase<SignatureTrainer>
165
165
/// <param name="featureColumn">Role name for features.</param>
166
166
/// <param name="labelColumn">Role name for label.</param>
167
167
/// <returns></returns>
168
- public static IDataTransform Create ( IHostEnvironment env , IDataView input , ITrainer trainer , string featureColumn = DefaultColumnNames . Features , string labelColumn = DefaultColumnNames . Label )
168
+ public static IDataTransform Create ( IHostEnvironment env ,
169
+ IDataView input ,
170
+ ITrainer trainer ,
171
+ string featureColumn = DefaultColumnNames . Features ,
172
+ string labelColumn = DefaultColumnNames . Label )
169
173
{
174
+ Contracts . CheckValue ( env , nameof ( env ) ) ;
170
175
var args = new Arguments ( )
171
176
{
172
177
FeatureColumn = featureColumn ,
@@ -178,16 +183,17 @@ public static IDataTransform Create(IHostEnvironment env, IDataView input, ITrai
178
183
179
184
public static IDataTransform Create ( IHostEnvironment env , Arguments args , IDataView input )
180
185
{
186
+ Contracts . CheckValue ( env , nameof ( env ) ) ;
187
+ env . CheckValue ( args , nameof ( args ) ) ;
188
+ env . CheckUserArg ( args . Trainer . IsGood ( ) , nameof ( args . Trainer ) ,
189
+ "Trainer cannot be null. If your model is already trained, please use ScoreTransform instead." ) ;
190
+
181
191
return Create ( env , args , args . Trainer . CreateInstance ( env ) , input ) ;
182
192
}
183
193
184
194
private static IDataTransform Create ( IHostEnvironment env , Arguments args , ITrainer trainer , IDataView input )
185
195
{
186
- Contracts . CheckValue ( env , nameof ( env ) ) ;
187
- env . CheckValue ( args , nameof ( args ) ) ;
188
196
env . CheckValue ( input , nameof ( input ) ) ;
189
- env . CheckUserArg ( args . Trainer . IsGood ( ) , nameof ( args . Trainer ) ,
190
- "Trainer cannot be null. If your model is already trained, please use ScoreTransform instead." ) ;
191
197
192
198
var host = env . Register ( "TrainAndScoreTransform" ) ;
193
199
0 commit comments