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