-
Notifications
You must be signed in to change notification settings - Fork 7
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
remove Tensorflow dependency #750
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #750 +/- ##
=======================================
Coverage 96.73% 96.73%
=======================================
Files 108 108
Lines 7498 7498
=======================================
Hits 7253 7253
Misses 245 245
Flags with carried forward coverage won't be shown. Click here to find out more.
|
pyproject.toml
Outdated
matplotlib = { version = "^3.7.0", optional = true } | ||
seaborn = { version = "^0.12.2", optional = true } | ||
pydot = { version = "^1.4.2", optional = true } | ||
tensorflow = { version = "^2.12.0", optional = true, markers = "sys_platform == 'linux' or sys_platform == 'darwin'" } | ||
# TODO: the marker is a temporary solution due to the lack of the tensorflow-io 0.32.0's wheels for Windows, this package is one of | ||
# the tensorflow requirements | ||
skl2onnx = { version = "^1.14.0", optional = true } | ||
onnxruntime = { version = "^1.14.1", optional = true } | ||
onnx = { version = "^1.13.1", optional = 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.
Do we still need onnx
? Is there a different way of serializing Scikit-Learn objects? (maybe even just dictionaries for the hyperparams, and dump in json)
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.
Thanks @Edoardo-Pedicillo, I only had that comment, and it could be split to a separate issue/PR as well. The PR itself seems alright.
Consider that the
At least not in But please, also check yourself. |
The
|
Correct, that's why I was not finding it in classifiers. So, you need of course to dump the result of the fit. But is there no intrinsic way of Scikit-Learn to serialize this? (I mean, w/o ONNX) |
Pikle ;) btw onnx is suggested by scikit itself (https://scikit-learn.org/stable/model_persistence.html) and even the other options require extra dependencies |
I was wondering if there was something like a There is also PMML (XML instead of binary) and Let's keep it as it is. |
This PR closes #734
Checklist:
master
main
main