distilbert-base-uncased_rotten_tomatoes
xianzhew
Clasificación de texto
Este modelo es una versión afinada de distilbert-base-uncased en el conjunto de datos de Rotten Tomatoes. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.9770, Precisión: 0.8405.
Como usar
El siguiente modelo se puede utilizar mediante los siguientes hiperparámetros durante el entrenamiento:
- learning_rate: 5e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- optimizer: Adam con betas=(0.9,0.999) y epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 5
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- TensorBoard
Casos de uso
- Clasificación de reseñas de películas
- Análisis de sentimientos en textos