Alireza1044/albert-base-v2-sst2

Alireza1044

Clasificación de texto

Este modelo es una versión afinada de albert-base-v2 en el conjunto de datos GLUE SST2. Alcanzó los siguientes resultados en el conjunto de evaluación: Pérdida: 0.3808, Precisión: 0.9232.

Como usar

{
  "learning_rate": 1e-05,
  "train_batch_size": 32,
  "eval_batch_size": 8,
  "seed": 42,
  "optimizer": "Adam con betas=(0.9,0.999) y epsilon=1e-08",
  "lr_scheduler_type": "lineal",
  "num_epochs": 4.0
}

Funcionalidades

Token separado: [SEP]
Token de relleno: [PAD]
Token de clase: [CLS]
Token de máscara: [MASK]
Pipeline de clasificación de texto
Compatible con PyTorch
Compatible con TensorBoard
Generado desde Trainer
Compatible con AutoTrain
Compatible con Endpoints

Casos de uso

Clasificación de texto
Análisis de sentimiento