Skip to content
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.

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

ONNX support. #785

Open
MuhammedSaleem-Brej opened this issue Aug 2, 2024 · 3 comments
Open

ONNX support. #785

MuhammedSaleem-Brej opened this issue Aug 2, 2024 · 3 comments

Comments

@MuhammedSaleem-Brej
Copy link

I hope to add ONNX format support to this excellent framework.
ONNX is an open-source and a standard format of AI models.
With ONNX we will be able to share our models between frameworks and programing languages.

@haifengl
Copy link
Owner

haifengl commented Aug 4, 2024

You are more than welcome to contribute to this project! I believe that there is already a Java binding for ONNX. Can you please share what you would like to develop? Thanks!

@MuhammedSaleem-Brej
Copy link
Author

Firstly, thank you very much for your generosity
And secondly I aim to make Smile framework able to import and export ONNX models using smile.io package.
The next virtual code may explain the idea.
import java.nio.file.Path;
import smile.classification.RandomForest;
import smile.io.Read;
import smile.io.Write;
// Import an ONNX model.
RandomForest model = (RandomForest) Read.onnx("/home/model.onnx");
// Export an ONNX model.
Write.onnx(model, Path.of("/home/model.onnx"));

@haifengl
Copy link
Owner

haifengl commented Aug 6, 2024

Cool. Feel free to create PR when you get it work. Thanks in advance!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants