README.md

metadata

language: en
license: mit
tags:
  - Text Classification

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)