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

Optimize scikit-learn Model Inference Using ONNX and Fix Missing Datasets in xgb_cpu_main_config.json for Benchmarking #181

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

JaimeAdanCuevas
Copy link

Description

Add a comprehensive description of proposed changes

List associated issue number(s) if exist(s): #6 (for example)

Documentation PR (if needed): #1340 (for example)


PR should start as a draft, then move to ready for review state after CI is passed and all applicable checkboxes are closed.
This approach ensures that reviewers don't spend extra time asking for regular requirements.

You can remove a checkbox as not applicable only if it doesn't relate to this PR in any way.

Checklist to comply with before moving PR from draft:

PR completeness and readability

  • I have reviewed my changes thoroughly before submitting this pull request.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have updated the documentation to reflect the changes or created a separate PR with update and provided its number in the description, if necessary.
  • Git commit message contains an appropriate signed-off-by string (see CONTRIBUTING.md for details).
  • I have added a respective label(s) to PR if I have a permission for that.
  • I have resolved any merge conflicts that might occur with the base branch.

Testing

  • I have run it locally and tested the changes extensively.
  • All CI jobs are green or I have provided justification why they aren't.
  • I have extended testing suite if new functionality was introduced in this PR.

Error handling (invalid JSON, missing files)
Automatic fixes (correct dataset paths if names mismatch)
Clear logging with warnings and actions
 Compares ONNX vs. TVM performance on a scikit-learn model
Demonstrates ONNX conversion from sklearn
Uses TVM to optimize the model for inference
Benchmarks inference times for performance analysis
Create refactoring_of_benchmarks.py
Create compilation_frameworks.py
@ethanglaser
Copy link
Contributor

Hi @JaimeAdanCuevas, it looks like there may be some additions needed before review. Could you please add a description and lint and let us know if you have any questions or need any support?

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

Successfully merging this pull request may close these issues.

2 participants