Added Propability to the output columns of binary

sfilipi · sfilipi · commit d47014e3ef32 · 2018-09-06T11:43:03.000-07:00
diff --git a/src/Microsoft.ML.StandardLearners/Standard/LinearPredictor.cs b/src/Microsoft.ML.StandardLearners/Standard/LinearPredictor.cs
@@ -487,7 +487,7 @@ protected override void SaveCore(ModelSaveContext ctx)
             ctx.SetVersionInfo(GetVersionInfo());
         }
 
-        public override PredictionKind PredictionKind => PredictionKind.BinaryClassification;        
+        public override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
 
         /// <summary>
         /// Combine a bunch of models into one by averaging parameters
diff --git a/src/Microsoft.ML.StandardLearners/Standard/Online/AveragedPerceptron.cs b/src/Microsoft.ML.StandardLearners/Standard/Online/AveragedPerceptron.cs
@@ -57,6 +57,7 @@ public AveragedPerceptronTrainer(IHostEnvironment env, Arguments args)
             OutputColumns = new[]
             {
                 new SchemaShape.Column(DefaultColumnNames.Score, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),
+                new SchemaShape.Column(DefaultColumnNames.Probability, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),
                 new SchemaShape.Column(DefaultColumnNames.PredictedLabel, SchemaShape.Column.VectorKind.Scalar, BoolType.Instance, false)
             };
         }
diff --git a/src/Microsoft.ML.StandardLearners/Standard/Online/LinearSvm.cs b/src/Microsoft.ML.StandardLearners/Standard/Online/LinearSvm.cs
@@ -92,6 +92,7 @@ public LinearSvm(IHostEnvironment env, Arguments args)
             OutputColumns = new[]
             {
                 new SchemaShape.Column(DefaultColumnNames.Score, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),
+                new SchemaShape.Column(DefaultColumnNames.Probability, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),
                 new SchemaShape.Column(DefaultColumnNames.PredictedLabel, SchemaShape.Column.VectorKind.Scalar, BoolType.Instance, false)
             };
         }

Original file line number	Diff line number	Diff line change
`@@ -487,7 +487,7 @@ protected override void SaveCore(ModelSaveContext ctx)`
`487`	`487`	`ctx.SetVersionInfo(GetVersionInfo());`
`488`	`488`	`}`
`489`	`489`
`490`		`- public override PredictionKind PredictionKind => PredictionKind.BinaryClassification;`
	`490`	`+ public override PredictionKind PredictionKind => PredictionKind.BinaryClassification;`
`491`	`491`
`492`	`492`	`/// <summary>`
`493`	`493`	`/// Combine a bunch of models into one by averaging parameters`
Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,7 @@ public AveragedPerceptronTrainer(IHostEnvironment env, Arguments args)`
`57`	`57`	`OutputColumns = new[]`
`58`	`58`	`{`
`59`	`59`	`new SchemaShape.Column(DefaultColumnNames.Score, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),`
	`60`	`+ new SchemaShape.Column(DefaultColumnNames.Probability, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),`
`60`	`61`	`new SchemaShape.Column(DefaultColumnNames.PredictedLabel, SchemaShape.Column.VectorKind.Scalar, BoolType.Instance, false)`
`61`	`62`	`};`
`62`	`63`	`}`
Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@ public LinearSvm(IHostEnvironment env, Arguments args)`
`92`	`92`	`OutputColumns = new[]`
`93`	`93`	`{`
`94`	`94`	`new SchemaShape.Column(DefaultColumnNames.Score, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),`
	`95`	`+ new SchemaShape.Column(DefaultColumnNames.Probability, SchemaShape.Column.VectorKind.Scalar, NumberType.R4, false),`
`95`	`96`	`new SchemaShape.Column(DefaultColumnNames.PredictedLabel, SchemaShape.Column.VectorKind.Scalar, BoolType.Instance, false)`
`96`	`97`	`};`
`97`	`98`	`}`