Skip to content

Latest commit

 

History

History
29 lines (20 loc) · 476 Bytes

README.md

File metadata and controls

29 lines (20 loc) · 476 Bytes
Raw

DRETC

1. Environments

  • python (3.6.8)
  • cuda (10.1)

2. Dependencies

  • numpy (1.17.3)
  • torch (1.6.0)
  • transformers (3.5.1)
  • pandas (1.1.5)
  • scikit-learn (0.23.2)

3. Preparation

  • Download DocRED dataset
  • Put all the train_annotated.json, dev.json, test.json,word2id.json,vec.npy,rel2id.json,ner2id into the directory data/
>> python preprocess.py

4. Run

>> python main.py