Yelp/yelp_review_full
Viewer โข Updated โข 700k โข 33.1k โข 156
How to use csabakecskemeti/bert-base-case-yelp5-tuned-experiment with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="csabakecskemeti/bert-base-case-yelp5-tuned-experiment") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("csabakecskemeti/bert-base-case-yelp5-tuned-experiment")
model = AutoModelForSequenceClassification.from_pretrained("csabakecskemeti/bert-base-case-yelp5-tuned-experiment", device_map="auto")bert-base-cased model fintuned with yelp5 dataset. One epoch on the full training set finetuning the full model and one more epoch finetune just the classifier layers on the full training set.