AmeliaCMU/AmeliaTF-weights-only

Overview

This repository hosts pre-trained weights for AmeliaTF.

What’s here

• weights/ — weights to use with the AmeliaTF repository on branch main.
• src-weights/ — weights to use with the AmeliaTF repository on branch main_src.

Both trees mirror the same structure:

• Multi-Airport/ — multi-airport training runs (e.g., seen-1, seen-2, …, seen-all).
• Single-Airport/ — per-airport training runs (e.g., kbos, kdca, kewr, …).
• Each run directory contains:
  • checkpoints/ — model checkpoint files (e.g., *.ckpt).
  • splits/ — train/val/test splits used for that run.
  • hydra/ — the Hydra configuration snapshot used for training (if present).

Examples

• Main branch weights:

  • weights/Single-Airport/kbos/checkpoints/kbos_critical.ckpt
  • weights/Multi-Airport/seen-all/checkpoints/seen-all.ckpt

• main_src branch weights:

  • src-weights/Single-Airport/kjfk/checkpoints/kjfk_random.ckpt
  • src-weights/Multi-Airport/seen-3/checkpoints/seen-3.ckpt

How to use

In your AmeliaTF codebase, select the appropriate branch and point your configuration or loading script to the desired checkpoint path from this repository.

High level guidance:

1. If you are on AmeliaTF branch main, use files under weights/.
2. If you are on AmeliaTF branch main_src, use files under src-weights/.
3. Choose the run you want (e.g., Single-Airport/kbos or Multi-Airport/seen-1) and load the .ckpt in its checkpoints/ folder.

Note: Exact loading commands depend on the version of AmeliaTF you are using. Refer to AmeliaTF’s documentation or training scripts for the model-loading API and expected checkpoint format.

Folder overview (truncated)

weights/
  Multi-Airport/
  Single-Airport/
src-weights/
  Multi-Airport/
  Single-Airport/

If you need additional metadata (training configs, metrics) not included here, check the hydra/ subfolders or contact the maintainers.