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

Fix issue with auto assignment with imbalanced classes #1067

Merged
merged 1 commit into from
Nov 15, 2023

Conversation

bloebp
Copy link
Member

@bloebp bloebp commented Nov 10, 2023

Before, the auto assignment threw an error when there are too few samples available from one class due to the cross validation. Now, using StratifiedKFold makes sure that the classes are balanced in the folds.

@bloebp bloebp requested a review from petergtz November 10, 2023 00:12
@bloebp bloebp force-pushed the fix_auto_assign_classif branch from 041d5ab to 0ed0efd Compare November 10, 2023 00:20
@bloebp bloebp force-pushed the fix_auto_assign_classif branch 2 times, most recently from 60a2c2e to f00efd2 Compare November 10, 2023 07:21
petergtz
petergtz previously approved these changes Nov 10, 2023
Base automatically changed from improve_kl_div to main November 15, 2023 15:00
@bloebp bloebp dismissed petergtz’s stale review November 15, 2023 15:00

The base branch was changed.

@bloebp bloebp force-pushed the fix_auto_assign_classif branch from f00efd2 to 34d3a87 Compare November 15, 2023 15:02
@bloebp bloebp merged commit b9ae10b into main Nov 15, 2023
30 checks passed
@bloebp bloebp deleted the fix_auto_assign_classif branch November 15, 2023 15:19
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