-
Notifications
You must be signed in to change notification settings - Fork 589
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
Rename SKLearnPredictor to Predictor #524
Conversation
…ex.ml.sklearn.Predictor and update tests.
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.
I agree with this plan.
packages/vaex-ml/vaex/ml/sklearn.py
Outdated
2 6.6 2.9 4.6 1.3 1 1.44276 | ||
''' | ||
|
||
warnings.warn(message='''This class is deprecated and it wil will be removed in vaex-ml 0.8. |
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.
If you put this at the classlevel, it will always be executed (at import time). Put this in the constructor, and remove all the docstrings, and pass arguments to super.
packages/vaex-ml/vaex/ml/sklearn.py
Outdated
from vaex.ml.state import serialize_pickle | ||
|
||
|
||
@vaex.serialize.register | ||
@generate.register | ||
class SKLearnPredictor(state.HasState): | ||
class Predictor(state.HasState): | ||
'''This class wraps any scikit-learn estimator (a.k.a predictor) making it a vaex pipeline object. | ||
|
||
By wrapping any scikit-learn estimators with this class, it becoes a vaex |
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.
By wrapping any scikit-learn estimators with this class, it becoes a vaex | |
By wrapping any scikit-learn estimators with this class, it becomes a vaex |
…ex.ml.sklearn.Predictor.
This PR renames
vaex.ml.sklearn.SKLearnPredictor
tovaex.ml.sklearn.Predictor
vaex.ml.sklearn.Predictor
. Implementation is identical, only the class name is changed.vaex.ml.sklearn.SKLearnPredictor
class and update the docstring to mention that the class is being deprecated.vaex.ml.sklearn.Predictor