autogluon / autogluon/autogluon

Model Serving - AutoGluonOnnx

Open
#1,079 9 comments 0 reactions 0 assignees View on GitHub
env: new help wanted question
Dominant language
Python
Stars
10.7k
Forks
1.2k
Avg merge
21h 29m
Merged PRs (30d)
57

Description

For tabular model, or any model class, is there any easy way or plan in the coming release to integrate with ONNX for downstream serving ?

A side question is for any individual trained model, e.g.`model = predictor._trainer.load_model('xgboost')`, is this directly compatible to things like mlflow.xgboost or bentoml.frameworks.xgboost.XgboostModelArtifact? It is not hard to write a customised wrapper, however would like to hear from your plan. I think it would be super good to consider the overall ecosystem when we have a such user friendly autogluon already.

Contributor guide

Open the contributing guide

Research direction

Start by reviewing the model-serving path around predictor._trainer.load_model('xgboost') and the requested ONNX, MLflow, and BentoML integrations. The issue names no files or tests; done would require a decided integration scope and a documented, tested serving path for the supported model classes.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.