CoolFace
Modelpublic

jordan2889/classify-phishing_real_1

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

classify-phishingreal1

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

  • —Loss: 0.1185
  • —Accuracy: 0.9645
  • —F1: 0.9645
  • —Precision: 0.9645
  • —Recall: 0.9645
  • —Accuracy Label 0: 0.9708
  • —Accuracy Label 1: 0.9559

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: 32
  • —evalbatchsize: 32
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 64
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 500
  • —num_epochs: 3

Training results

Training LossEpochStepValidation LossAccuracyF1PrecisionRecallAccuracy Label 0Accuracy Label 1
0.49910.10301000.47480.79250.78190.81360.79250.95080.5747
0.30870.20602000.30520.87990.87930.87990.87990.91890.8262
0.29740.30903000.23900.90930.90940.90950.90930.91810.8972
0.26440.41194000.30680.86630.86700.88870.86630.78990.9715
0.2230.51495000.21220.91540.91580.91950.91540.89050.9495
0.2150.61796000.20110.92290.92220.92520.92290.97140.8561
0.14190.72097000.18360.93050.93000.93180.93050.96900.8775
0.15110.82398000.18280.93050.93080.93270.93050.91450.9526
0.1730.92699000.15440.94300.94280.94330.94300.96660.9107
0.09861.029910000.15130.94290.94300.94350.94290.93840.9491
0.14031.132911000.15150.94260.94290.94440.94260.92780.9631
0.11331.235812000.13940.94750.94750.94750.94750.95310.9397
0.11171.338813000.15250.94570.94590.94670.94570.93710.9576
0.12771.441814000.13110.94900.94910.94920.94900.95010.9475
0.08861.544815000.13750.95030.95030.95030.95030.96280.9331
0.12731.647816000.12970.95330.95330.95350.95330.95360.9529
0.11021.750817000.11360.95780.95780.95780.95780.96370.9498
0.07931.853818000.12690.95620.95610.95630.95620.97180.9348
0.09951.956719000.11290.95910.95900.95910.95910.97020.9437
0.08462.059720000.13620.95330.95340.95430.95330.94220.9685
0.0962.162721000.13830.95630.95640.95720.95630.94670.9696
0.07972.265722000.11370.96200.96190.96190.96200.97110.9494
0.06022.368723000.12110.96090.96090.96090.96090.96640.9532
0.09512.471724000.11940.96140.96150.96150.96140.96280.9596
0.03432.574725000.12370.96290.96290.96300.96290.96240.9634
0.05122.677726000.12630.96250.96250.96250.96250.97380.9471
0.05322.780627000.12290.96330.96330.96330.96330.97060.9533
0.06732.883628000.12060.96440.96440.96440.96440.96790.9596
0.02092.986629000.11850.96450.96450.96450.96450.97090.9556

Framework versions

  • —Transformers 4.42.3
  • —Pytorch 2.2.1
  • —Datasets 2.20.0
  • —Tokenizers 0.19.1