Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

textattack
/
albert-base-v2-ag-news

Text Classification
Transformers
PyTorch
albert
Model card Files Files and versions
xet
Community
2

Instructions to use textattack/albert-base-v2-ag-news with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use textattack/albert-base-v2-ag-news with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="textattack/albert-base-v2-ag-news")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("textattack/albert-base-v2-ag-news")
    model = AutoModelForSequenceClassification.from_pretrained("textattack/albert-base-v2-ag-news")
  • Notebooks
  • Google Colab
  • Kaggle
New discussion
Resources
  • PR & discussions documentation
  • Code of Conduct
  • Hub documentation

What are the labels? "Label_0" does not really mean anything!

#2 opened over 2 years ago by
ekoerner

Adding `safetensors` variant of this model

#1 opened almost 3 years ago by
SFconvertbot
Company
TOS Privacy About Careers
Website
Models Datasets Spaces Pricing Docs