Skip to content
This repository has been archived by the owner on Sep 13, 2023. It is now read-only.

mlem.api.save: ML frameworks and Data formats support #423

Open
17 of 22 tasks
aguschin opened this issue Oct 3, 2022 · 4 comments
Open
17 of 22 tasks

mlem.api.save: ML frameworks and Data formats support #423

aguschin opened this issue Oct 3, 2022 · 4 comments
Labels
data-format Data format support ml-framework ML Framework support ✨ epic High-level task that should be split into small ones

Comments

@aguschin
Copy link
Contributor

aguschin commented Oct 3, 2022

This is an epic to collect all ML frameworks / Data formats we would like to support, when you call mlem.api.save(mymodel, "model") or mlem.api.save(mydataset, "dataset").

ML frameworks:

Data formats:

  • Simple Python types (str, int, float, dict, etc)
  • Pandas
  • Numpy
  • Lightgbm
  • XGBoost

Please feel free to post a comment if you need something we don't support yet :)

@aguschin aguschin added ml-framework ML Framework support data-format Data format support ✨ epic High-level task that should be split into small ones labels Oct 3, 2022
@aguschin aguschin pinned this issue Oct 3, 2022
@aguschin aguschin changed the title save: ML frameworks and Data formats support mlem.api.save: ML frameworks and Data formats support Oct 3, 2022
@madhur-tandon
Copy link
Contributor

we support lightgbm as well, updated the description.

@Snehashish-s
Copy link

I am getting a "mlem.core.errors.HookNotFound: No suitable ModelHook for object of type "IsolationForest" " exception when saving Isolationforest model, any way it can be solved ?

@aguschin
Copy link
Contributor Author

Hi @Snehashish-s, rn MLEM doesn't support this, but it's pretty trivial to add by adding OutlierMixin to this line https://github.com/iterative/mlem/blob/main/mlem/contrib/sklearn.py#L31. Would you like contribute this and add a test?

@aguschin
Copy link
Contributor Author

aguschin commented Jul 13, 2023

@Snehashish-s, I just supported IsolationForest in https://github.com/iterative/mlem/releases/tag/0.4.14

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
data-format Data format support ml-framework ML Framework support ✨ epic High-level task that should be split into small ones
Projects
None yet
Development

No branches or pull requests

3 participants