CoolFace
Modelpublic

Mediocre-Judge/my_mind_classifier_BertMini

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes13downloads
README.md69 linesDownload Raw Back to root
1---2library_name: transformers3license: mit4base_model: prajjwal1/bert-mini5tags:6- generated_from_trainer7metrics:8- accuracy9- precision10- recall11- f112model-index:13- name: my_mind_classifier_BertMini14  results: []15---16 17<!-- This model card has been generated automatically according to the information the Trainer had access to. You18should probably proofread and complete it, then remove this comment. -->19 20# my_mind_classifier_BertMini21 22This model is a fine-tuned version of [prajjwal1/bert-mini](https://huggingface.co/prajjwal1/bert-mini) on the None dataset.23It achieves the following results on the evaluation set:24- Loss: 0.453025- Accuracy: 0.862026- Precision: 0.857827- Recall: 0.862028- F1: 0.859929 30## Model description31 32More information needed33 34## Intended uses & limitations35 36More information needed37 38## Training and evaluation data39 40More information needed41 42## Training procedure43 44### Training hyperparameters45 46The following hyperparameters were used during training:47- learning_rate: 2e-0548- train_batch_size: 6449- eval_batch_size: 6450- seed: 4251- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0852- lr_scheduler_type: linear53- num_epochs: 254 55### Training results56 57| Training Loss | Epoch | Step  | Validation Loss | Accuracy | Precision | Recall | F1     |58|:-------------:|:-----:|:-----:|:---------------:|:--------:|:---------:|:------:|:------:|59| 0.6179        | 1.0   | 9024  | 0.5588          | 0.8303   | 0.8272    | 0.8303 | 0.8288 |60| 0.4973        | 2.0   | 18048 | 0.4530          | 0.8620   | 0.8578    | 0.8620 | 0.8599 |61 62 63### Framework versions64 65- Transformers 4.45.166- Pytorch 2.4.067- Datasets 3.0.168- Tokenizers 0.20.069