Skip to content

chore: default to stratified_group kfold everywhere - #21

Open
vojtech-cifka wants to merge 1 commit into
masterfrom
chore/unify-kfold-convention
Open

vojtech-cifka wants to merge 1 commit into
masterfrom
chore/unify-kfold-convention

Conversation

@vojtech-cifka

Copy link
Copy Markdown
Collaborator

Point both kfold run ids at the splits recomputed over the corrected filter_tiles output (#20).

Make stratified_group the default in both base tasks and drop the now redundant per-experiment overrides, so one run id drives every experiment. All nine ml experiments resolve to the same kfold run id, verified by composing each config. Plain stratified folds remain available by overriding kfold_strategy and kfold_run_id together.

Point both kfold run ids at the splits recomputed over the corrected
filter_tiles output (#20).

The kfold convention was inconsistent: the base tasks defaulted to plain
stratified, while train_*_group_kfold and final_linear_provgigapath_*
overrode to stratified_group. final_linear_virchow2_* overrode nothing, so
it read stratified_kfold_run_id while its probe used the group split.

That mismatch did not change the final training data — final_embedding_tiles
sets neither include_folds nor exclude_folds, so the final fit uses every row
of kfold_tiles.parquet and ignores the fold column, and both artifacts hold
the same tile set. It mattered because the two run ids have to be kept
current independently: the stratified one still pointed at a split built over
the pre-fix filter_tiles, so the final virchow2 fit would have trained on the
old tile set, missing the 30% of annotated tiles the scale bug dropped. The
run also recorded a kfold run id its probe never used.

Make stratified_group the default in both base tasks and drop the now
redundant per-experiment overrides, so one run id drives every experiment.
All nine ml experiments resolve to the same kfold run id, verified by
composing each config. Plain stratified folds remain available by overriding
kfold_strategy and kfold_run_id together.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
@vojtech-cifka
vojtech-cifka requested review from a team, JakubPekar and matejpekar October 8, 2026 16:48
@vojtech-cifka vojtech-cifka self-assigned this Oct 8, 2026
@coderabbitai

coderabbitai Bot commented Oct 8, 2026

Copy link
Copy Markdown

Important

  • 🔍 Trigger review

This repository does not receive automatic reviews because it has fewer than 10 stars.

⚙️ Run configuration
  • Configuration used: defaults
  • Review profile: CHILL
  • Plan: Advanced
  • Run ID: d33f1318-7755-4dc1-b316-d4ffbc8bbc81
  • Autopilot · Keep fixing CodeRabbit findings and required CI, and resolving merge conflicts

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

@vojtech-cifka
vojtech-cifka requested review from vejtek and removed request for JakubPekar October 8, 2026 16:48
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.

3 participants