cybersectony commited on
Commit
e94292b
1 Parent(s): aafe3ac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -17,7 +17,7 @@ tags:
17
  # A distilBERT based Phishing Email Detection Model
18
 
19
  ## Model Overview
20
- This model is specifically fine-tuned for detecting phishing emails using the Hugging Face Trainer API.
21
 
22
  ## Key Specifications
23
  - __Base Architecture:__ DistilBERT
@@ -33,7 +33,8 @@ This model is specifically fine-tuned for detecting phishing emails using the Hu
33
 
34
  ## Dataset Details
35
 
36
- This model was trained using a [Phishing Email Detection Dataset](https://huggingface.co/datasets/cybersectony/PhishingEmailDetection).
 
37
 
38
  ## Usage Guide
39
 
 
17
  # A distilBERT based Phishing Email Detection Model
18
 
19
  ## Model Overview
20
+ This model is based on DistilBERT and has been fine-tuned for multilabel classification of Emails and URLs as safe or potentially phishing.
21
 
22
  ## Key Specifications
23
  - __Base Architecture:__ DistilBERT
 
33
 
34
  ## Dataset Details
35
 
36
+ The model was trained on a custom dataset of Emails and URLs labeled as legitimate or phishing. The dataset is available at [`cybersectony/PhishingEmailDetection`](https://huggingface.co/datasets/cybersectony/PhishingEmailDetection) on the Hugging Face Hub.
37
+
38
 
39
  ## Usage Guide
40