# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("dvs/poetry-author")
model = AutoModelForSequenceClassification.from_pretrained("dvs/poetry-author")Quick Links
Model Trained Using AutoTrain
- Problem type: Text Classification
Validation Metrics
loss: 6.5507588386535645
f1_macro: 0.007553613837545024
f1_micro: 0.04596774193548386
f1_weighted: 0.02342410713485415
precision_macro: 0.0064405619945614515
precision_micro: 0.04596774193548387
precision_weighted: 0.020352886732684962
recall_macro: 0.017759772448305164
recall_micro: 0.04596774193548387
recall_weighted: 0.04596774193548387
accuracy: 0.04596774193548387
- Downloads last month
- 3
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="dvs/poetry-author")