hipom_data_mapping/train/README.md

28 lines
767 B
Markdown
Raw Normal View History

# 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.
The folders are named with the following convention:
`<method_type>`\_`<model_type>`\_`<prediction_type>`\_`<included_fields>`
e.g.
"classification_bert_complete_desc_unit",
which means: folder to perform classification using the bert model, predicting
for the complete thing+property output using description and unit
To train, just run `python train.py`
The inference code is within a folder `classification_prediction` or
`mapping_prediction`.
Note: the classification\_t5 folders are depracated in favor of the BERT-based
classification models.