Downloads · 30 days
166
16% of all-time downloads
drug-combo-classifier-pubmedbert-dapt
allenai/drug-combo-classifier-pubmedbert-dapt
drug-combo-classifier-pubmedbert-dapt is a machine learning model from allenai. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This is the baseline model used in most experiments in the paper "A Dataset for N-ary Relation Extraction of Drug Combinations".
All-time downloads
1.1K
Public
Repo size
4.1 GB
Likes
5
Public
Repo makeup
Click a slice to open those files.
.chkpt682 MB · 61%
At a glance
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- May 4, 2022
- Updated
- Oct 20, 2022
- SHA
- ac15c668
Model card
From the Hugging Face model README
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
- 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
- Download model from Huggingface:
git lfs install
git clone https://huggingface.co/allenai/drug-combo-classifier-pubmedbert-dapt
- Load model (
in Python):
from modeling.model import load_model
checkpoint_path = "drug-combo-classifier-pubmedbert-dapt"
model, tokenizer, metadata = load_model(checkpoint_path)
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- May 4, 2022
- Updated
- Oct 20, 2022