Skip to content

fix: correct examples configuration and fix validation data never being checked#91

Merged
meilame-tayebjee merged 6 commits intomainfrom
julber/wip
May 7, 2026
Merged

fix: correct examples configuration and fix validation data never being checked#91
meilame-tayebjee merged 6 commits intomainfrom
julber/wip

Conversation

@julber95
Copy link
Copy Markdown
Collaborator

@julber95 julber95 commented May 7, 2026

Fix several bugs in the examples scripts and core training logic:

  • Set raw_labels=False in all TrainingConfig calls since labels are already integer-encoded

  • Set raw_categorical_inputs=False in train and predict calls since categorical features are already integer-encoded

  • Remove check_val_every_n_epoch=2 which caused ModelCheckpoint to never save (val_loss unavailable on skipped epochs), leading to a crash on load_from_checkpoint

  • Fix X_val always being None, causing validation data to be silently ignored during training

@julber95 julber95 changed the title fix: correct examples configuration fix: correct examples configuration and fix validation data never being checked May 7, 2026
@meilame-tayebjee meilame-tayebjee self-requested a review May 7, 2026 16:35
@meilame-tayebjee
Copy link
Copy Markdown
Member

Thanks !

@meilame-tayebjee meilame-tayebjee merged commit f269062 into main May 7, 2026
5 checks passed
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