CoolFace
Modelpublic

bkissinger10/cfd_model2

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes9downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

cfd_model2

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

  • Loss: 0.0066
  • Precision: 0.9978
  • Recall: 0.9987
  • F1: 0.9982
  • Accuracy: 0.9986

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: 3e-05
  • trainbatchsize: 4
  • evalbatchsize: 4
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • num_epochs: 4

Training results

Training LossEpochStepValidation LossPrecisionRecallF1Accuracy
0.04930.191000.03520.98900.99300.99100.9937
0.01030.392000.02730.98860.99560.99210.9946
0.00110.583000.01900.99210.99780.99500.9960
0.02470.774000.01670.99780.99430.99600.9968
0.00030.975000.02690.99260.99780.99520.9961
0.00361.166000.01330.99600.99600.99600.9968
0.00081.357000.02220.99260.99870.99560.9965
0.00031.558000.02870.98950.99740.99340.9953
0.00051.749000.01320.99340.99820.99580.9970
0.00241.9310000.01230.99520.99820.99670.9977
0.00072.1311000.00990.99690.99430.99560.9965
0.00052.3212000.00870.99780.99650.99710.9977
0.0012.5113000.00550.99740.99910.99820.9986
0.00022.7114000.00490.99740.99820.99780.9986
0.00042.915000.00650.99690.99820.99760.9984
0.00023.0916000.00710.99690.99780.99740.9982
0.00013.2917000.00770.99740.99780.99760.9984
0.00023.4818000.00720.99740.99780.99760.9984
0.00053.6819000.00720.99740.99780.99760.9984
0.02073.8720000.00660.99780.99870.99820.9986

Framework versions

  • Transformers 4.36.2
  • Pytorch 2.1.2+cu121
  • Datasets 2.14.6
  • Tokenizers 0.15.0