allenai/drug-combo-classifier-pubmedbert-dapt

This is the baseline model used in most experiments in the paper "A Dataset for N-ary Relation Extraction of Drug Combinations".

(for just the domain-adapted masked language model that we use underneath this model, see here)

Steps to load this model

1. Download accompanying code:

git clone https://github.com/allenai/drug-combo-extraction.git
conda create --name drug_combo python=3.8.5
conda activate drug_combo

2. Download model from Huggingface:

git lfs install
git clone https://huggingface.co/allenai/drug-combo-classifier-pubmedbert-dapt

3. Load model (in Python):

from modeling.model import load_model
checkpoint_path = "drug-combo-classifier-pubmedbert-dapt"
model, tokenizer, metadata = load_model(checkpoint_path)