workaround Scores

dotnet · Nov 11, 2019 · d7f9996 · d7f9996
1 parent 7fba31c
commit d7f9996
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Microsoft.ML.Data/Scorers/SchemaBindablePredictorWrapper.cs b/src/Microsoft.ML.Data/Scorers/SchemaBindablePredictorWrapper.cs
@@ -320,7 +320,7 @@ private protected override bool SaveAsOnnxCore(OnnxContext ctx, RoleMappedSchema
             if (!ctx.ContainsColumn(featName))
                 return false;
             Contracts.Assert(ctx.ContainsColumn(featName));
-            return mapper.SaveAsOnnx(ctx, new[] { outputNames[1] }, ctx.GetVariableName(featName));
+            return mapper.SaveAsOnnx(ctx, outputNames, ctx.GetVariableName(featName));
         }
 
         private protected override ISchemaBoundMapper BindCore(IChannel ch, RoleMappedSchema schema) =>