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
Added lightgbm integration #544
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just some tiny things here FWIW.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a few typos, apart from that it look good to merge from my side 👍
docs/book/features/integrations.md
Outdated
@@ -41,6 +41,7 @@ These are the third-party integrations that ZenML currently supports: | |||
| Hugging Face | ✅ | Materializer | Use Hugging Face tokenizers, datasets and models. | [huggingface](https://github.com/zenml-io/zenml/tree/main/examples/huggingface) | | |||
| KServe | ⛏ | Inference | Looking for community implementors. | | | |||
| Kubeflow | ✅ | Orchestrator | Either full Kubeflow or Kubeflow Pipelines. Works for local environments currently. | [kubeflow](https://github.com/zenml-io/zenml/tree/main/examples/kubeflow) | | |||
| lightgbm | ✅ | Training | Support for `Booster` and `Dataset` materialization. | [xgboost](https://github.com/zenml-io/zenml/tree/main/examples/lightgbm) | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
still has some xgboost references in there
|
||
# Make a temporary phantom artifact | ||
with tempfile.NamedTemporaryFile( | ||
mode="w", suffix=".json", delete=True |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not that it will break anything, but I'm a little confused why the suffix is .json
here and the default filename is model.txt
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Its a copy pasta! Fixed now!
Co-authored-by: Alex Strick van Linschoten <strickvl@users.noreply.github.com>
Co-authored-by: Alex Strick van Linschoten <strickvl@users.noreply.github.com>
Co-authored-by: Michael Schuster <schustmi@users.noreply.github.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM 🚀
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Describe changes
I added lightgbm integration as i thought it was a quick win after xgboost. Its basically the same with some caveats as #538!
Pre-requisites
Please ensure you have done the following:
Types of changes