Suicidal-BERT

Este modelo de clasificación de texto predice si una secuencia de palabras es suicida (1) o no suicida (0). Fue entrenado en el conjunto de datos de Suicide and Depression obtenido de Kaggle, que fue recopilado de Reddit y consta de 232,074 filas distribuidas equitativamente entre 2 clases: suicidio y no suicidio. El ajuste fino del modelo se realizó en 1 época, con un tamaño de lote de 6 y una tasa de aprendizaje de 0.00001. Debido a recursos de cómputo limitados y tiempo, no pudimos escalar el número de épocas y el tamaño de los lotes.