Team Ai
Modelpublic

ml4pubmed/scibert-scivocab-uncased_pub_section

sourceHugging Faceupdated 3y agoView on Hugging Face
1likes17downloads
Model Card

scibert-scivocab-uncasedpubsection

  • —original model file name: textclassiferscibertscivocabuncasedpubmed_full
  • —This is a fine-tuned checkpoint of allenai/scibert_scivocab_uncased for document section text classification
  • —possible document section classes are:BACKGROUND, CONCLUSIONS, METHODS, OBJECTIVE, RESULTS,

usage in python

install transformers as needed: pip install -U transformers

run the following, changing the example text to your use case:

from transformers import pipeline

model_tag = "ml4pubmed/scibert-scivocab-uncased_pub_section"
classifier = pipeline(
              'text-classification', 
              model=model_tag, 
            )
            
prompt = """
Experiments on two machine translation tasks show these models to be superior in quality while being more parallelizable and requiring significantly less time to train.
"""

classifier(
    prompt,
) # classify the sentence

metadata

training_metrics

  • —daterun: Apr-25-2022t-03
  • —huggingfacetag: allenai/scibertscivocab_uncased

training_parameters

  • —daterun: Apr-25-2022t-03
  • —huggingfacetag: allenai/scibertscivocab_uncased