arcade-training-scripts / train_reranker.py

Commit History

Eval and save every 25 steps for detailed curves
a21d6c7
verified

amanwithaplan commited on

Increase epochs to 10
1b22ef9
verified

amanwithaplan commited on

Replace correlation with proper ranking metrics (NDCG, MRR)
cddbc43
verified

amanwithaplan commited on

Add domain metrics callback + fix push_to_hub exist_ok
e76af51
verified

amanwithaplan commited on

Fix: select only required columns for training
a6f0d21
verified

amanwithaplan commited on

Add trackio metrics and correlation evaluation
55c1fce
verified

amanwithaplan commited on

Upload train_reranker.py with huggingface_hub
3517d13
verified

amanwithaplan commited on