shawhin
/

bert-phishing-classifier_student

Text Classification

Model card Files Files and versions Community

shawhin commited on Aug 27, 2024

Commit

0963214

·

verified ·

1 Parent(s): b08e59b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ pipeline_tag: text-classification
 This model is modified version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) trained via knowledge distillation from [shawhin/bert-phishing-classifier_teacher](https://huggingface.co/shawhin/bert-phishing-classifier_teacher) using the [shawhin/phishing-site-classification]() dataset.
 It achieves the following results on the testing set:
-- Loss: 0.0563
 - Accuracy: 0.9022
 - Precision: 0.9426
 - Recall: 0.8603

 This model is modified version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) trained via knowledge distillation from [shawhin/bert-phishing-classifier_teacher](https://huggingface.co/shawhin/bert-phishing-classifier_teacher) using the [shawhin/phishing-site-classification]() dataset.
 It achieves the following results on the testing set:
+- Loss (training): 0.0563
 - Accuracy: 0.9022
 - Precision: 0.9426
 - Recall: 0.8603