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

mlflow-package should have ready examples for predictions for pytorch, tensorflow (both mnist) and xgboost (covertype) #140

Open
Zethson opened this issue Dec 1, 2020 · 2 comments
Assignees
Labels
enhancement New feature or request low priority

Comments

@Zethson
Copy link
Member

Zethson commented Dec 1, 2020

Currently the mlflow-packages templates for pytorch, tensorflow and xgboost do not really have out of the box working predictions.
Only xgboost works, but it is not using the covertype example and therefore does not match with the general mlflow-xgboost model.

mlflow-package for pytorch should load a trained mnist model and perform a prediction. Maybe even plot one thing.
Same for Tensorflow (mnist) and XGBoost with covertype.

@Zethson
Copy link
Member Author

Zethson commented Feb 2, 2021

@emiller88 assigned to you as well for now. This one will come last and will require some more explanation from me.

@Zethson
Copy link
Member Author

Zethson commented Mar 19, 2021

@Zethson Let's use a minimal Poetry for this one.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request low priority
Projects
None yet
Development

No branches or pull requests

2 participants