Richard Wong
1f3970459f
Feat: introduced BERT-based binary classification |
||
|---|---|---|
| .. | ||
| classification_bert_complete_desc | ||
| classification_bert_complete_desc_unit | ||
| classification_bert_pattern_desc | ||
| classification_bert_pattern_desc_unit | ||
| classification_t5_complete | ||
| classification_t5_complete_with_constrastive | ||
| classification_t5_mdm_with_contrastive | ||
| mapping_baseline | ||
| mapping_t5_complete_desc_unit | ||
| mapping_t5_pattern_desc_unit | ||
| 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.
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.