|
--- |
|
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) |
|
``` |
