Skip to content

Feature/implement ray - #6

Merged
Sam-v6 merged 11 commits into
mainfrom
feature/implement-ray
Nov 19, 2025
Merged

Sam-v6 merged 11 commits into
mainfrom
feature/implement-ray

Conversation

@Sam-v6

@Sam-v6 Sam-v6 commented Nov 19, 2025 •

Copy link
Copy Markdown
Owner

What does this PR do?

Adding Ray Tune for hyper-parameter optimization

What are some specific details on the changes?

  • Paths
    • Replacing os.path.join() with path objects
    • Replacing environment.sh with paths.py which bring in PROJECT_ROOT and SAVE_DIR
  • Ray Tune
    • Bringing in Ray Tune with MLflow callback and checkpoint dir
    • Allowing for optional tune run to do hyper-param optimization with --tune
    • Made training more generic to allow HPO to run over top of it if desired
  • Miscl
    • Making timing decorator off by default, can be called in training with --profile
    • Getting rid of un-used files
    • Adding generate week 9 predictions to the pipeline

How has this been tested?

Tune is not tested in pipeline, need to implement conditional checks later

Important Notes

None

Checklist

  • Tests have been added or updated to cover the changes.
  • Documentation has been updated (if necessary)

@Sam-v6
Sam-v6 merged commit d7395d2 into main Nov 19, 2025
1 check passed
@Sam-v6
Sam-v6 deleted the feature/implement-ray branch November 19, 2025 04:25
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