Richard Wong
22429ea536
Feat: added mapping to pattern-only method Chore: re-organized prediction to be within mapping folders |
||
|---|---|---|
| .. | ||
| classification_all | ||
| classification_all_with_contrastive | ||
| classification_mdm_with_contrastive | ||
| mapping_baseline | ||
| mapping_pattern | ||
| README.md | ||
README.md
Train
What is this folder
Here contains the code for training and mapping evaluation.
Each folder contains a training variation.
After training, each folder contains the checkpoint files for each fold.
mapping directory contains the code to run the model on test data and also
produce the csv outputs.