Codebase for Stat 288 project on IWI prediction with vision transformers.
Dataset images-2014 not included, but can be downloaded with data_download_script.py and Static Maps API.
Model.py runs fine-tuning on both CNN and ViT.
Attention.py visualizes attention matrices and input overlays from model checkpoints.