From e631e321d3e5fef84c2d2e93368cf9de2535fad3 Mon Sep 17 00:00:00 2001 From: Gani Nazirov Date: Wed, 5 Feb 2020 12:13:39 -0800 Subject: [PATCH] Fix onnx output for GcnTransform --- src/Microsoft.ML.Transforms/GcnTransform.cs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/Microsoft.ML.Transforms/GcnTransform.cs b/src/Microsoft.ML.Transforms/GcnTransform.cs index 9bd4abe6ee..09ad97db5d 100644 --- a/src/Microsoft.ML.Transforms/GcnTransform.cs +++ b/src/Microsoft.ML.Transforms/GcnTransform.cs @@ -617,7 +617,8 @@ public void SaveAsOnnx(OnnxContext ctx) continue; } - if (!SaveAsOnnxCore(ctx, iinfo, ctx.GetVariableName(inputColumnName), ctx.AddIntermediateVariable(_srcTypes[iinfo], inputColumnName))) + string outputColumnName = _parent.ColumnPairs[iinfo].outputColumnName; + if (!SaveAsOnnxCore(ctx, iinfo, ctx.GetVariableName(inputColumnName), ctx.AddIntermediateVariable(_srcTypes[iinfo], outputColumnName))) { ctx.RemoveColumn(inputColumnName, true); }