@@ -25,7 +25,7 @@ public void TrainerGeneratorBasicNamedParameterTest()
25
25
Pipeline pipeline = new Pipeline ( new PipelineNode [ ] { node } ) ;
26
26
CodeGenerator codeGenerator = new CodeGenerator ( pipeline , ( null , null ) ) ;
27
27
var actual = codeGenerator . GenerateTrainerAndUsings ( ) ;
28
- string expected = "LightGbm(learningRate:0.1f,numLeaves:1,labelColumn:\" Label\" ,featureColumn:\" Features\" ); " ;
28
+ string expected = "LightGbm(learningRate:0.1f,numLeaves:1,labelColumn:\" Label\" ,featureColumn:\" Features\" )" ;
29
29
Assert . AreEqual ( expected , actual . Item1 ) ;
30
30
Assert . IsNull ( actual . Item2 ) ;
31
31
}
@@ -45,7 +45,7 @@ public void TrainerGeneratorBasicAdvancedParameterTest()
45
45
Pipeline pipeline = new Pipeline ( new PipelineNode [ ] { node } ) ;
46
46
CodeGenerator codeGenerator = new CodeGenerator ( pipeline , ( null , null ) ) ;
47
47
var actual = codeGenerator . GenerateTrainerAndUsings ( ) ;
48
- string expectedTrainer = "LightGbm(new Options(){LearningRate=0.1f,NumLeaves=1,UseSoftmax=true,LabelColumn=\" Label\" ,FeatureColumn=\" Features\" }); " ;
48
+ string expectedTrainer = "LightGbm(new Options(){LearningRate=0.1f,NumLeaves=1,UseSoftmax=true,LabelColumn=\" Label\" ,FeatureColumn=\" Features\" })" ;
49
49
string expectedUsing = "using Microsoft.ML.LightGBM;\r \n " ;
50
50
Assert . AreEqual ( expectedTrainer , actual . Item1 ) ;
51
51
Assert . AreEqual ( expectedUsing , actual . Item2 ) ;
@@ -163,7 +163,7 @@ public void TrainerComplexParameterTest()
163
163
Pipeline pipeline = new Pipeline ( new PipelineNode [ ] { node } ) ;
164
164
CodeGenerator codeGenerator = new CodeGenerator ( pipeline , ( null , null ) ) ;
165
165
var actual = codeGenerator . GenerateTrainerAndUsings ( ) ;
166
- string expectedTrainer = "LightGbm(new Options(){Booster=new TreeBooster(){},LabelColumn=\" Label\" ,FeatureColumn=\" Features\" }); " ;
166
+ string expectedTrainer = "LightGbm(new Options(){Booster=new TreeBooster(){},LabelColumn=\" Label\" ,FeatureColumn=\" Features\" })" ;
167
167
var expectedUsings = "using Microsoft.ML.LightGBM;\r \n " ;
168
168
Assert . AreEqual ( expectedTrainer , actual . Item1 ) ;
169
169
Assert . AreEqual ( expectedUsings , actual . Item2 ) ;
0 commit comments