Commit Graph

4 Commits

Author SHA1 Message Date
Michael Pilosov
e9d3d72499 Improve training signal and add honest eval metrics
- prepare_training_data: bag_size 5→8, street signs fill slots first so
  every sample contains the most geographically discriminative texts
- train: HuberLoss replaces MSE (robust to outlier intersections),
  ReduceLROnPlateau scheduler added, split logic extracted to data_utils
- eval: reproduce train/val split to report honest per-bag and
  per-intersection-aggregated metrics separately for train and val sets
- data_utils: shared split_indices() so train and eval use identical splits

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-25 22:41:14 +00:00
Michael Pilosov
40967a303c drop pena 2026-05-25 22:09:57 +00:00
Michael Pilosov
b5810dd282 more training data + frozen layers options 2026-05-25 15:16:19 -06:00
Michael Pilosov
e8419354f4 first train/eval 2026-05-25 14:11:05 -06:00