-
Notifications
You must be signed in to change notification settings - Fork 34
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
Add XGBoost JSON Reader/writer #461
Conversation
a rebase should handle the test failures now |
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.
looks good! just one question
) -> Artifact: | ||
"""Log an XGBoost model as a JSON file to this client object. | ||
|
||
Please note that we do not currently support logging directly from the SKLearn interface. |
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.
what do you mean by this?
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 users use the XGBRegressor
or XGBClassifier
classes that implement sklearn interface for xgboost this will not work. They would have to use model.get_booster()
first and log that. An obvious addition is to check for that ourselves and then call model.get_booster()
if relevant.
I can add that in this PR if you want.
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.
ahh gotcha, nah this is fine for now
closing and re-opening to try to get the license check to work |
Related to #458
What
How to Test