Skip to content

Conversation

Innixma
Copy link
Contributor

@Innixma Innixma commented Feb 7, 2025

Issue #, if available:

Resolves #4869

Description of changes:

For the 1.2 release, I refactored the sanity checks in BaggedEnsembleModel, and as part of it, I made it raise an exception if called with k_fold=0, as previously we passed both k_fold=0 and k_fold=1 to mean the same thing (doing refit_full). However, this actually made refit_full fail to work for models anymore. This wasn't caught in our unit tests because the fallback mechanism which uses the first fold model in case of an error made the unit tests continue to pass.

I've fixed the issue by passing k_fold=1 instead of k_fold=0 during refit_full.

I've also added a docstring to clarify this behavior in BaggedEnsembleModel, and a unit test to verify that it is working correctly.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Copy link

github-actions bot commented Feb 7, 2025

Job PR-4870-0bb52bb is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4870/0bb52bb/index.html

@Innixma Innixma merged commit 55c3d9f into autogluon:master Feb 11, 2025
26 checks passed
Copy link

Job PR-4870-71bdbf7 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4870/71bdbf7/index.html

@Innixma Innixma deleted the fix_refit_full branch April 16, 2025 21:23
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.

[tabular] refit_full show unnecessary error warning
2 participants