File size: 504 Bytes
10cfebe 1e1f3b4 10cfebe 58db556 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
---
language: en
license: mit
---
How to load model:
git clone https://github.com/allenai/drug-combo-extraction.git
conda create --name drug_combo python=3.8.5
conda activate drug_combo
cd drug-combo-extraction
# download checkpoints_pubmedbert_cpt_2021_three_class.tgz
tar -xvzf checkpoints_pubmedbert_cpt_2021_three_class.tgz
python
> from modeling.model import load_model
> checkpoint_path = "checkpoints_pubmedbert_cpt_2021_three_class"
> model, tokenizer, metadata = load_model(checkpoint_path)
|