level2-klue-nlp-03

🌱Members

김민혁	김의진	김성우	오원택	정세연

Relation Extraction

Relation Extraction is the task of predicting attributes and relations between subject and object entity in sentence.

Getting started

# Install Packages
pip install -r requirements.txt

# Train & Inference
python src/train.py

📂Structure

root/
|
|-- config.yaml
|-- config_sweep.yaml
|
|-- src/
|   |-- train.py
|   |-- train_sweep.py
|   |-- loader.py
|   |-- models.py
|   |-- utils.py
|
|-- eda/
|   |-- entity_EDA.ipynb
|   |-- result_eda.ipynb

Data

Augmentation

• Easy Data Augmentation
• Back Translation

Preprocessing

• Hanja Removal
• Unidecode

Model

• Entity Marker
• Binary Classifier
• RECENT(entity relation restriction)
• Additional Model for Fine-tuning