SetFit/sst2
Viewer • Updated • 9.61k • 6.81k • 11
How to use Islamamro/yelp-sentiment-aurora-islamamro with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Islamamro/yelp-sentiment-aurora-islamamro") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Islamamro/yelp-sentiment-aurora-islamamro")
model = AutoModelForSequenceClassification.from_pretrained("Islamamro/yelp-sentiment-aurora-islamamro", device_map="auto")Text classifier fine-tuned from microsoft/MiniLM-L12-H384-uncased on SetFit/sst2 (2 classes).
Built, trained and published end-to-end via the Aurora Research Portal by islamamro. Trained on CPU on a 1200-example demo subset.
microsoft/MiniLM-L12-H384-uncased (~33M parameters)This is a demonstration of the Aurora build -> train -> publish pipeline on a small subset, not a production model. Model size here is set by the base architecture, which is varied deliberately across the Aurora showcase to span a range of parameter counts. For production, fine-tune on the full dataset.
Base model
microsoft/MiniLM-L12-H384-uncased