Averaged Perceptron pipeline serialization fix (dotnet#257)

daholste · Dmitry-A · commit 370286739623 · 2019-08-22T14:40:15.000-07:00
diff --git a/src/Microsoft.ML.Auto/TrainerExtensions/BinaryTrainerExtensions.cs b/src/Microsoft.ML.Auto/TrainerExtensions/BinaryTrainerExtensions.cs
@@ -51,7 +51,7 @@ public PipelineNode CreatePipelineNode(IEnumerable<SweepableParam> sweepParams,
             {
                 additionalProperties = new Dictionary<string, object>()
                 {
-                    { "NumberOfIterations", DefaultNumIterations.ToString() }
+                    { "NumberOfIterations", DefaultNumIterations }
                 };
             }
 
diff --git a/src/Test/TrainerExtensionsTests.cs b/src/Test/TrainerExtensionsTests.cs
@@ -183,7 +183,7 @@ public void BuildDefaultAveragedPerceptronPipelineNode()
   ],
   ""Properties"": {
     ""LabelColumn"": ""L"",
-    ""NumberOfIterations"": ""10""
+    ""NumberOfIterations"": 10
   }
 }";
             Util.AssertObjectMatchesJson(expectedJson, pipelineNode);

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ public PipelineNode CreatePipelineNode(IEnumerable<SweepableParam> sweepParams,`
`51`	`51`	`{`
`52`	`52`	`additionalProperties = new Dictionary<string, object>()`
`53`	`53`	`{`
`54`		`- { "NumberOfIterations", DefaultNumIterations.ToString() }`
	`54`	`+ { "NumberOfIterations", DefaultNumIterations }`
`55`	`55`	`};`
`56`	`56`	`}`
`57`	`57`
Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ public void BuildDefaultAveragedPerceptronPipelineNode()`
`183`	`183`	`],`
`184`	`184`	`""Properties"": {`
`185`	`185`	`""LabelColumn"": ""L"",`
`186`		`- ""NumberOfIterations"": ""10""`
	`186`	`+ ""NumberOfIterations"": 10`
`187`	`187`	`}`
`188`	`188`	`}";`
`189`	`189`	`Util.AssertObjectMatchesJson(expectedJson, pipelineNode);`