Skip to content

Latest commit

 

History

History
33 lines (21 loc) · 951 Bytes

README.md

File metadata and controls

33 lines (21 loc) · 951 Bytes
Raw

This code is for paper "Support-Query Prototype Guidance and Enhanced Few-Shot Relational Triple Extraction"

Requirements

python=3.7
pytorch=1.9.1
cuda=10.2
transformers=2.8.0

Usage

  1. Training

For example, if you want to train our SQGE in 5-way-5-shot setting, it should be like:

python main.py --model=proto_dot --trainN=5 --evalN=5 --K=5 --Q=1 --O=0 --distance_metric="conv" --contrastive="Normal" --temperature_alpha=0.5 --temperature_beta=0.1 --alpha=0.5 --beta=0.5 --threshold="None" --result_dir="result/proto_hcl"
  1. Testing

For example, if you want to test our SQGE in 5-way-5-shot setting, it should be like:

python main.py --model=proto_dot --trainN=5 --evalN=5 --K=5 --Q=1 --O=0 --distance_metric="conv" --contrastive="Normal" --temperature_alpha=0.5 --temperature_beta=0.1 --alpha=0.5 --beta=0.5 --threshold="None" --load_ckpt="proto_dot_fewrel_xxxx" --result_dir="result/proto_hcl"