Skip to content
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 Sklearn API #68

Closed
wants to merge 10 commits into from
Closed

Conversation

joshdunnlime
Copy link

@joshdunnlime joshdunnlime commented Dec 3, 2023

[x] add sklearn regressor based on XGBRegressor
[x] add some general tests based on tests_model
[x] move the optional extra SHap package imports to where the function is called*

[x] improve test coverage
[x] add documentation
[x] examples regressor

[ ] improve documentation

  • the sklearn module imports from the model module. This seems to throw an ImportError for SHap when not installed.

@joshdunnlime
Copy link
Author

@StatMixedML The majority of functionality available in XGBoost and XGBoostLSS is now available in the XGBoostLSS Sklearn API. I think now would be a great time to get a bit of a review, even if just high-level thought and comments.

@StatMixedML
Copy link
Owner

@joshdunnlime Can you please move the PR to branch sklearn_integration, thanks.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants