Skip to content

Commit

Permalink
Multiple inputs output support for OnnxTransform. (#1586)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
jignparm authored Nov 20, 2018
1 parent 3ce839e commit dafa30c
Show file tree
Hide file tree
Showing 5 changed files with 420 additions and 231 deletions.
10 changes: 5 additions & 5 deletions src/Microsoft.ML.OnnxTransform/OnnxCatalog.cs
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,13 @@ public static class OnnxCatalog
/// </summary>
/// <param name="catalog">The transform's catalog.</param>
/// <param name="modelFile">The path of the file containing the ONNX model.</param>
/// <param name="inputColumn">The input column.</param>
/// <param name="outputColumn">The output column resulting from the transformation.</param>
/// <param name="inputColumns">The input columns.</param>
/// <param name="outputColumns">The output columns resulting from the transformation.</param>
public static OnnxScoringEstimator ApplyOnnxModel(this TransformsCatalog catalog,
string modelFile,
string inputColumn,
string outputColumn)
=> new OnnxScoringEstimator(CatalogUtils.GetEnvironment(catalog), modelFile, inputColumn, outputColumn);
string[] inputColumns,
string[] outputColumns)
=> new OnnxScoringEstimator(CatalogUtils.GetEnvironment(catalog), modelFile, inputColumns, outputColumns);

/// <summary>
/// Initializes a new instance of <see cref="OnnxScoringEstimator"/>.
Expand Down
Loading

0 comments on commit dafa30c

Please sign in to comment.