Text Classification
Transformers
PyTorch
bert
Generated from Trainer
Eval Results (legacy)
text-embeddings-inference
Instructions to use Etelis/YELP_BERT_5E with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Etelis/YELP_BERT_5E with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Etelis/YELP_BERT_5E")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Etelis/YELP_BERT_5E") model = AutoModelForSequenceClassification.from_pretrained("Etelis/YELP_BERT_5E") - Notebooks
- Google Colab
- Kaggle
Asking about the dataset you are using
#3
by f6fg1vt - opened
Your readme said that you use "full_review" dataset, however, when I print the "mode" ,it shows me that you are using "polarity" one since your output only contains 2 dimension.Could you explain that to me?
Hey, this is an auto generated description, it was trained on a filtered (binary) YELP dataset :)
Good catch tho' :)