MMG code is available at https://github.com/MWXGOD/MMG
- python (3.8.12)
- cuda (11.4)
- numpy (1.21.4)
- torch (1.10.0)
- gensim (4.1.2)
- transformers (4.13.0)
- pandas (1.3.4)
- scikit-learn (1.0.1)
- prettytable (2.4.0)
We provide some datasets processed in our code.
>> python main-run-start-end-mul-out-1-new-decoding.py --config ./config/BC5CDR.json --shot 5-1 --learning_rate 0.003 --epochs 20 --loss2_weight '0.2' --loss3_weight '0.2'