mental-bert-base-uncased-finetuned-0505

YeRyeongLee/mental-bert-base-uncased-finetuned-0505

mental-bert-base-uncased-finetuned-0505 is a text classification model from YeRyeongLee. Use it when you need a label for a piece of text. It is set up for transformers.

should probably proofread and complete it, then remove this comment. --

Downloads · 30 days

1% of all-time downloads

All-time downloads

1.5K

Public

Repo size

1.3 GB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.bin438 MB · 100%

At a glance

Task: Text Classification
Library: transformers
Model type: bert
Access: Public
Created: May 5, 2022
Updated: May 5, 2022
SHA: 78a2f58e

Try a prompt

Model card

From the Hugging Face model README

mental-bert-base-uncased-finetuned-0505

This model is a fine-tuned version of mental/mental-bert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:

Loss: 0.4195
Accuracy: 0.9181
F1: 0.9182

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 2e-05
train_batch_size: 8
eval_batch_size: 8
seed: 42
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 4

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy	F1
No log	1.0	1373	0.2846	0.9124	0.9119
No log	2.0	2746	0.3468	0.9132	0.9129
No log	3.0	4119	0.3847	0.9189	0.9192
No log	4.0	5492	0.4195	0.9181	0.9182

Framework versions

Transformers 4.11.3
Pytorch 1.9.0
Datasets 1.16.1
Tokenizers 0.10.3

Task: Text Classification
Library: transformers
Type: bert
Created: May 5, 2022
Updated: May 5, 2022