CoolFace
Modelpublic

viperDEE/phishing-links-detection-using-transformers

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
1likes12downloads
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. -->

distilbert-zim-phishing

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

  • Loss: 0.3919
  • Accuracy: 0.9032
  • Precision: 0.8667
  • Recall: 0.9286
  • F1: 0.8966

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: 4
  • evalbatchsize: 8
  • seed: 42
  • optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 20
  • num_epochs: 10
  • mixedprecisiontraining: Native AMP
  • labelsmoothingfactor: 0.1

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1
0.61241.0270.56580.751.00.42860.6
0.53542.0540.27461.01.01.01.0
0.27723.0810.20401.01.01.01.0
0.39674.01080.20371.01.01.01.0
0.33365.01350.22891.01.01.01.0
0.20116.01620.20001.01.01.01.0
0.19957.01890.20191.01.01.01.0
0.19978.02160.20251.01.01.01.0
0.35129.02430.20191.01.01.01.0
0.314210.02700.20221.01.01.01.0

Framework versions

  • Transformers 5.0.0
  • Pytorch 2.10.0+cu128
  • Datasets 4.0.0
  • Tokenizers 0.22.2