fancyzhx/ag_news
Viewer • Updated • 128k • 112k • 190
How to use Alesteba/deep_model_09_clasificador-news-2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Alesteba/deep_model_09_clasificador-news-2") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Alesteba/deep_model_09_clasificador-news-2")
model = AutoModelForSequenceClassification.from_pretrained("Alesteba/deep_model_09_clasificador-news-2")This model is a fine-tuned version of albert-base-v2 on the ag_news dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.6332 | 1.0 | 715 | 0.4676 | 0.8812 |
| 0.5148 | 2.0 | 1430 | 0.4496 | 0.9006 |
| 0.3638 | 3.0 | 2145 | 0.4530 | 0.9033 |