distilbert-finetuned-ag-news
MuntasirHossain
Clasificación de texto
Este modelo es una versión afinada de distilbert-base-uncased en el conjunto de datos de AG News. Este es un modelo DistilBERT afinado en el conjunto de datos AG News para determinar si un artículo de noticias está dentro de alguna de las cuatro categorías: 'Mundo', 'Deportes', 'Negocios' y 'Ciencia/Tecnología'. La API Trainer se utilizó para entrenar el modelo.
Como usar
Para usar este modelo, puedes seguir estos pasos:
```python
from transformers import pipeline
distilbert_finetuned_ag_news = pipeline("text-classification", model="MuntasirHossain/distilbert-finetuned-ag-news")
result = distilbert_finetuned_ag_news("Input your text here.")
print(result)
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- Generado con Trainer
Casos de uso
- Clasificación de artículos de noticias en 'Mundo', 'Deportes', 'Negocios' y 'Ciencia/Tecnología'
- Analizar la distribución de categorías de noticias
- Soporte a sistemas de noticias automatizadas