juliensimon/reviews-sentiment-analysis
juliensimon
Clasificación de texto
Modelo DistilBERT afinado en reseñas de productos en inglés. Un cuaderno para Amazon SageMaker está disponible en la subcarpeta 'code'.
Como usar
import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name = 'juliensimon/reviews-sentiment-analysis'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
# Tokenización de una reseña de ejemplo
evaluation_text = 'Este producto es excelente!'
tokens = tokenizer(evaluation_text, return_tensors='pt')
# Predicción
outputs = model(**tokens)
probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
print(probs)
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- Safetensors
- Inferencia de análisis de sentimientos
Casos de uso
- Análisis de sentimientos en reseñas de productos
- Evaluación de feedback de clientes
- Despliegue en puntos de inferencia dedicados