SkyAsl
/

Bert-base-Comment_classifier

Text Classification

Model card Files Files and versions

Model Card for Model ID

Training Details

Training Data

https://huggingface.co/datasets/stanfordnlp/imdb

Training Hyperparameters

-learning_rate = 2e-4 -batch_size = 128 -num_epochs = 3 -weight_decay = 0.01

Evaluation

---Training_loss: 0.355100 ---Validation_loss: 0.330199 ---Accuracy: 0.857 ---Roc_auc: 0.936

Downloads last month: 7

Safetensors

Model size

0.1B params

Tensor type

F32

·

Model tree for SkyAsl/Bert-base-Comment_classifier

Base model

google-bert/bert-base-uncased

Finetuned

(6267)

this model

Dataset used to train SkyAsl/Bert-base-Comment_classifier