AshtonIsNotHere's picture
Update README.md
5454ef0
|
raw
history blame
641 Bytes
metadata
language:
  - en
tags:
  - medical
pipeline_tag: fill-mask

GatorTron-OG-breast-cancer

GatorTron-OG domain adapted on a set of breast cancer studies from the U.S. National Library of Medicine meeting the following criteria:

  • study is an interventional trial
  • study status is listed as completed
from transformers import AutoModel, AutoTokenizer

model_name = "AshtonIsNotHere/GatorTron-OG-breast-cancer"
tokenizer = AutoTokenizer.from_pretrained(
    model_name,
    padding="max_length",
    truncation=True,
)
model = AutoModelForMaskedLM.from_pretrained(
    model_name
)