shawhin commited on
Commit
6b2f5a9
1 Parent(s): 49b644e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -3
README.md CHANGED
@@ -1,3 +1,39 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - shawhin/phishing-site-classification
5
+ metrics:
6
+ - accuracy
7
+ - recall
8
+ - precision
9
+ - f1
10
+ base_model: shawhin/bert-phishing-classifier_student
11
+ library_name: transformers
12
+ ---
13
+
14
+
15
+ # bert-phishing-classifier_student
16
+
17
+ This model is a quantized version of [shawhin/bert-phishing-classifier_student](https://huggingface.co/shawhin/bert-phishing-classifier_student) trained via knowledge distillation from [shawhin/bert-phishing-classifier_teacher](https://huggingface.co/shawhin/bert-phishing-classifier_teacher) using the [shawhin/phishing-site-classification](https://huggingface.co/datasets/shawhin/phishing-site-classification) dataset.
18
+ It achieves the following results on the validation set:
19
+
20
+ - Accuracy: 0.9356
21
+ - Precision: 0.9757
22
+ - Recall: 0.8933
23
+ - F1 Score: 0.9327
24
+
25
+ ## Model description
26
+
27
+ Quantized student model for model compression example.
28
+
29
+ Video: coming soon! <br>
30
+ Blog: coming soon! <br>
31
+ Example code: coming soon!
32
+
33
+ ## Intended uses & limitations
34
+
35
+ This model was created for educational purposes.
36
+
37
+ ## Training and evaluation data
38
+
39
+ The Training, Testing, and Validation data are available here: [shawhin/phishing-site-classification](https://huggingface.co/datasets/shawhin/phishing-site-classification).