Richard Wong
59bbf1f403
Feat: implement bert classification |
||
|---|---|---|
| .. | ||
| classification_all | ||
| classification_all_with_contrastive | ||
| classification_bert | ||
| 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.