Skip to content

Add DataViewSchema overloads to ConvertToOnnx #6449

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
FranklinWhale wants to merge 18 commits into
base: main
Choose a base branch
from
Open

Add DataViewSchema overloads to ConvertToOnnx #6449

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
4735447
Add `DataViewSchema` overloads
FranklinWhale Nov 12, 2022
8ca34dd
Delete extra whitespaces
FranklinWhale Nov 12, 2022
3e8349b
Try custom EmptyDataView
FranklinWhale Nov 12, 2022
f412ab0
Don't use `record`
FranklinWhale Nov 12, 2022
daacf2e
Delete extra space
FranklinWhale Nov 12, 2022
ba3aaef
Fix errors
FranklinWhale Nov 12, 2022
7de47ee
Back to built-in `EmptyDataView`
FranklinWhale Nov 12, 2022
0180ff6
Test build failure
FranklinWhale Nov 13, 2022
3c111b9
Try again
FranklinWhale Nov 13, 2022
04d43e2
Change `ConvertToOnnxProtobuf`
FranklinWhale Nov 13, 2022
070d159
Add console log message for debug
FranklinWhale Nov 13, 2022
2317305
Delete log in test framework
FranklinWhale Nov 14, 2022
77a4fea
Add log in EmptyDataView
FranklinWhale Nov 14, 2022
a1a59fd
Fix formatting
FranklinWhale Nov 14, 2022
bf3ed06
Try DataFrame
FranklinWhale Nov 14, 2022
db3a2b5
Change log location
FranklinWhale Nov 14, 2022
33ef982
Clean up for review
FranklinWhale Nov 14, 2022
aed491f
Correct `catelog` param doc
FranklinWhale Nov 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Try again
  • Loading branch information
@FranklinWhale
FranklinWhale authored Nov 13, 2022
commit 3c111b91942fef531e453a43afd81ef7d09f5824
6 changes: 3 additions & 3 deletions src/Microsoft.ML.OnnxConverter/OnnxExportExtensions.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ internal static ModelProto ConvertToOnnxProtobuf(this ModelOperationsCatalog cat
/// <param name="stream">The stream to write the protobuf model to.</param>
/// <returns>An ONNX model equivalent to the converted ML.NET model.</returns>
public static void ConvertToOnnx(this ModelOperationsCatalog catalog, ITransformer transform, IDataView inputData, Stream stream) =>
ConvertToOnnxProtobuf(catalog, transform, inputData).WriteTo(stream);
ConvertToOnnxProtobuf(catalog, transform, new EmptyDataView(catalog.GetEnvironment(), inputData.Schema)).WriteTo(stream);

/// <summary>
/// Convert the specified <see cref="ITransformer"/> to ONNX format and writes to a stream.
Expand All @@ -91,7 +91,7 @@ public static void ConvertToOnnx(this ModelOperationsCatalog catalog, ITransform
/// <param name="stream">The stream to write the protobuf model to.</param>
/// <returns>An ONNX model equivalent to the converted ML.NET model.</returns>
public static void ConvertToOnnx(this ModelOperationsCatalog catalog, ITransformer transform, IDataView inputData, int opSetVersion, Stream stream) =>
ConvertToOnnxProtobuf(catalog, transform, inputData, opSetVersion).WriteTo(stream);
ConvertToOnnxProtobuf(catalog, transform, new EmptyDataView(catalog.GetEnvironment(), inputData.Schema), opSetVersion).WriteTo(stream);

/// <summary>
/// Convert the specified <see cref="ITransformer"/> to ONNX format and writes to a stream.
Expand All @@ -103,7 +103,7 @@ public static void ConvertToOnnx(this ModelOperationsCatalog catalog, ITransform
/// <param name="outputColumns">List of output columns we want to keep.</param>
/// <returns>An ONNX model equivalent to the converted ML.NET model.</returns>
public static void ConvertToOnnx(this ModelOperationsCatalog catalog, ITransformer transform, IDataView inputData, Stream stream, params string[] outputColumns) =>
ConvertToOnnxProtobuf(catalog, transform, inputData, outputColumns).WriteTo(stream);
ConvertToOnnxProtobuf(catalog, transform, new EmptyDataView(catalog.GetEnvironment(), inputData.Schema), outputColumns).WriteTo(stream);

/// <summary>
/// Convert the specified <see cref="ITransformer"/> to ONNX format and writes to a stream.
Expand Down