Skip to content

Add surrogate training - #2

Merged
Crivella merged 21 commits into
AI-TranspWood:mainfrom
Crivella:feature-surrogate_training
Oct 5, 2026
Merged

Crivella merged 21 commits into
AI-TranspWood:mainfrom
Crivella:feature-surrogate_training

Conversation

@Crivella

@Crivella Crivella commented Sep 22, 2026 •

Copy link
Copy Markdown
Member

Changes

  • Add training logic for the surrogate model
    • Include logic for transfer-learning (start from pre-trained model and run with frozen layers)
    • Include logic for cross-validation learning (uses a KFold split of a single set of data to automatically generate the train/validation/test subsets)
  • Add wood_ms train CLI command to start a surrogate training

Refactor

  • Split pipeline logic into a separate class

@Crivella
Crivella force-pushed the feature-surrogate_training branch from 19b9417 to 7e1ebab Compare September 24, 2026 08:52
@Crivella
Crivella force-pushed the feature-surrogate_training branch from 7e1ebab to fe76c70 Compare September 24, 2026 09:30
@Crivella
Crivella marked this pull request as ready for review September 24, 2026 11:08
@Crivella Crivella changed the title WIP: Add surrogate training Add surrogate training Sep 24, 2026
@Crivella
Crivella merged commit aecde62 into AI-TranspWood:main Oct 5, 2026
3 checks passed
@Crivella
Crivella deleted the feature-surrogate_training branch October 5, 2026 12:34
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.

1 participant