# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("alex2awesome/quote-type-prediction__basic")
model = AutoModelForSequenceClassification.from_pretrained("alex2awesome/quote-type-prediction__basic")Quick Links
Simple quote-type prediction model, based on AutoModelSequenceClassification.
Scores: "test_loss": 0.8912712335586548, "test_macro_f1": 0.5103889782929176, "test_micro_f1": 0.7168432590427475,
Which is slightly lower than this one: https://huggingface.co/alex2awesome/quote-type-sentence-model Which is trained on the same dataset.
Usage:
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="alex2awesome/quote-type-prediction__basic")