-
Notifications
You must be signed in to change notification settings - Fork 187
[ENH] scikit-learn compatible GAMRegressor and GAMClassifier classes
#364
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
base: main
Are you sure you want to change the base?
Conversation
….gitignore, and implement tests for GAMRegressor and GAMClassifier
…s and restore test_sklearn_api.py
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #364 +/- ##
==========================================
- Coverage 95.00% 94.81% -0.20%
==========================================
Files 22 24 +2
Lines 3202 3411 +209
==========================================
+ Hits 3042 3234 +192
- Misses 160 177 +17 ☔ View full report in Codecov by Sentry. |
…ssifier to support categorical features
… remove main execution block
fkiraly
left a comment
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.
This is excellent, thanks!
Are you still working on this? If so, it would be great if you could use parametrize_with_checks to test the sklearn estimator, this runs an entire battery of tests.
scikit-learn compatible GAMRegressor and GAMClassifier classes
scikit-learn compatible GAMRegressor and GAMClassifier classesscikit-learn compatible GAMRegressor and GAMClassifier classes
Description
This pull request introduces two new classes,
GAMRegressorandGAMClassifier, to the sklearn_api.py file. These classes provide scikit-learn compatible models using Generalized Additive Models (GAM) from the pygam library. The integration allows seamless use of GAM within scikit-learn's estimator interface, enabling these models to be used in standard machine learning pipelines.Changes
GAMClassifierclass to support classification tasks using GAM.__main__block with synthetic data generation, model fitting, and evaluation.Example Usage