Team Ai
Modelpublic

yimiwang/bert-petco-filtered_fontsize-ctr

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes23downloads
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. -->

bert-petco-filtered_fontsize-ctr

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

  • —Loss: 0.0022
  • —Mse: 0.0022
  • —Rmse: 0.0464
  • —Mae: 0.0364
  • —R2: 0.4468
  • —Accuracy: 0.8

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

Training results

Training LossEpochStepValidation LossMseRmseMaeR2Accuracy
0.04031.0150.01190.01190.10910.0921-2.05640.3
0.01672.0300.00310.00310.05550.04490.21000.6
0.01253.0450.00370.00370.06100.05140.04290.5167
0.00984.0600.00260.00260.05080.04070.33770.6333
0.01055.0750.00290.00290.05430.04130.24390.7
0.00836.0900.00290.00290.05350.03920.26440.7
0.00527.01050.00260.00260.05120.04320.32800.7
0.00448.01200.00230.00230.04820.03760.40250.7667
0.00459.01350.00370.00370.06050.05190.06010.4333
0.003310.01500.00280.00280.05250.03670.29090.7
0.003511.01650.00240.00240.04870.04030.39180.7667
0.00312.01800.00220.00220.04710.03670.43050.75
0.002313.01950.00220.00220.04640.03640.44680.8
0.002214.02100.00240.00240.04890.03870.38630.7167
0.002515.02250.00240.00240.04920.03790.37920.75
0.002116.02400.00250.00250.04970.03580.36670.7833
0.002117.02550.00240.00240.04860.03660.39430.7667
0.001718.02700.00240.00240.04850.03540.39500.7833
0.001719.02850.00280.00280.05290.03880.28190.75
0.001820.03000.00250.00250.04960.03630.36760.7667

Framework versions

  • —Transformers 4.38.2
  • —Pytorch 2.2.1+cu121
  • —Datasets 2.18.0
  • —Tokenizers 0.15.2