Skip to content

Commit dafa30c

Browse files
authored
Multiple inputs output support for OnnxTransform. (#1586)
* Added multiple input/output support for Onnxtransform * Comment out multi-output test until model is ready * addressed PR comments * update command line test to allow multi input/outputs * PR comments, plus activated 1 extra unit test for multi-in-out-puts * create tensor from span instead of array * small change to kick off build * remove commented code
1 parent 3ce839e commit dafa30c

File tree

5 files changed

+420
-231
lines changed

5 files changed

+420
-231
lines changed

src/Microsoft.ML.OnnxTransform/OnnxCatalog.cs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,13 @@ public static class OnnxCatalog
1515
/// </summary>
1616
/// <param name="catalog">The transform's catalog.</param>
1717
/// <param name="modelFile">The path of the file containing the ONNX model.</param>
18-
/// <param name="inputColumn">The input column.</param>
19-
/// <param name="outputColumn">The output column resulting from the transformation.</param>
18+
/// <param name="inputColumns">The input columns.</param>
19+
/// <param name="outputColumns">The output columns resulting from the transformation.</param>
2020
public static OnnxScoringEstimator ApplyOnnxModel(this TransformsCatalog catalog,
2121
string modelFile,
22-
string inputColumn,
23-
string outputColumn)
24-
=> new OnnxScoringEstimator(CatalogUtils.GetEnvironment(catalog), modelFile, inputColumn, outputColumn);
22+
string[] inputColumns,
23+
string[] outputColumns)
24+
=> new OnnxScoringEstimator(CatalogUtils.GetEnvironment(catalog), modelFile, inputColumns, outputColumns);
2525

2626
/// <summary>
2727
/// Initializes a new instance of <see cref="OnnxScoringEstimator"/>.

0 commit comments

Comments
 (0)