# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ml4pubmed/albert-base-v2_pub_section")
model = AutoModelForSequenceClassification.from_pretrained("ml4pubmed/albert-base-v2_pub_section")Quick Links
albert-base-v2_pub_section
- original model file name: textclassifer_albert-base-v2_pubmed_full
- This is a fine-tuned checkpoint of
albert-base-v2for document section text classification - possible document section classes are:BACKGROUND, CONCLUSIONS, METHODS, OBJECTIVE, RESULTS,
metadata
training_parameters
date_run: Apr-26-2022_t-04
huggingface_tag: albert-base-v2
- Downloads last month
- 6
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ml4pubmed/albert-base-v2_pub_section")