Skip to content

Comments

Fix estimator type#5

Open
jonathan-taylor wants to merge 197 commits intojonathan-taylor:masterfrom
intro-stat-learning:fix_estimator_type
Open

Fix estimator type#5
jonathan-taylor wants to merge 197 commits intojonathan-taylor:masterfrom
intro-stat-learning:fix_estimator_type

Conversation

@jonathan-taylor
Copy link
Owner

sklearn has introduced the sklearn_tags() method. this change uses the functions is_regressor/is_classifier instead of checking private attribute.

jonathan-taylor and others added 30 commits January 18, 2024 11:53
Co-authored-by: Jonathan Taylor <jonathan.taylor@stanford.edu>
See: pandas-dev/pandas#53805

Passing html literal strings is deprecated.

Wrap literal string/bytes input in ``io.StringIO``/``io.BytesIO`` instead.
* test of datasets

* docstring fix, fix labels of confusion matrix

* flexible level for CIs in summarize
* improved plot for 2-class SVM

* added margin as well
Adds the `__sklearn_tags__` method to the `sklearn_sm` and `sklearn_selected` wrappers. This allows scikit-learn to correctly identify the estimator type (regressor or classifier) based on the statsmodels model.

This change enables the use of scikit-learn's cross-validation and model selection tools with these wrappers.

Tests have been added to verify that OLS and GLM Binomial models are correctly identified.
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.

5 participants