1.7 KiB

Raw Permalink Blame History

Easy Start

English | 简体中文

Requirements

python == 3.8

torch == 1.5
transformers == 3.4.0
hydra-core == 1.0.6
deepke

Download Code

git clone https://github.com/zjunlp/DeepKE.git
cd DeepKE/example/re/few-shot

Install with Pip

Create and enter the python virtual environment.
Install dependencies: pip install -r requirements.txt.

Train and Predict

Dataset
- Download the dataset to this directory.
```
wget 120.27.214.45/Data/re/few-shot/data.tar.gz
tar -xzvf data.tar.gz
```
- The dataset SEMEVAL is stored in data:
  - rel2id.json：Relation Label - ID
  - temp.txt：Results of handled relation labels
  - test.txt： Test set
  - train.txt: Training set
  - val.txt：Validation set
Training
- Parameters, model paths and configuration for training are in the conf folder and users can modify them before training.
- Few-shot training on SEMEVAL
```
python run.py
```
- The trained model is stored in the current directory by default.
- Start to train from last-trained model
  
  modify train_from_saved_model in .yaml as the path of the last-trained model
- Logs for training are stored in the current directory by default and the path can be configured by modifying log_dir in .yaml
Prediction
```
python predict.py
```

Model