Implement train test split - #37

Merged

SarahAlidoost merged 17 commits into

mainfrom

Apr 28, 2026

Commits on Apr 13, 2026

calculate stats on train/test split
SarahAlidoost
committed
2a205fc
fix predict for train/test split
SarahAlidoost
committed
ceda34b
return loss in predict
SarahAlidoost
committed
f4e2c69

Commits on Apr 14, 2026

fix minor things
SarahAlidoost
committed
dd14702
add set_seed to utils
SarahAlidoost
committed
8da363f
expose dropout, replace Adam with AdamW optimizer
SarahAlidoost
committed
7bc7a4b
add validation, refactor
SarahAlidoost
committed
613f039
add gap loss between train and validation set
SarahAlidoost
committed
949a89d
fix linter errors
SarahAlidoost
committed
2688d8d

Commits on Apr 20, 2026

add stride to dataset, fix calculating stats
SarahAlidoost
committed
69868ad
fix compute_masked_loss support for nan in target
SarahAlidoost
committed
0a5c3a0
rerun nb
SarahAlidoost
committed
1a2fb22

Commits on Apr 22, 2026

Update climanet/train.py

SarahAlidoost
and
rogerkuou
authored
7d0f99f

Commits on Apr 28, 2026

bring the nb from main
SarahAlidoost
committed
0ca5a6f
Merge branch 'main' into train_test
SarahAlidoost
committed
b1b7df2
fix minor thing
SarahAlidoost
committed
4348879
bring the nb from branch
SarahAlidoost
committed
d996236