Downloads · 30 days
15
30% of all-time downloads
malicious-url-detection
UICHEOL-HWANG/malicious-url-detection
malicious-url-detection is a text classification model from UICHEOL-HWANG. Use it when you need a label for a piece of text. It is set up for transformers.
All-time downloads
50
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- distilbert
- Access
- Public
- Created
- Feb 12, 2025
- Updated
- Feb 17, 2025
- SHA
- c86577a8
Try a prompt
Base models
Model card
From the Hugging Face model README
Dacon 악성 URL 분류 AI 경진대회
링크
base_model : distilbert/distilbert-base-uncased
tune parameters
batch_size : 64
epocs : 5
learning_rate : 2e-5
weight_decay : 0.01
Model Training Results
| Step | Training Loss | Validation Loss | Accuracy | F1 Score |
|---|---|---|---|---|
| 200 | 0.024300 | 0.453651 | 0.929750 | 0.928860 |
| 400 | 0.016700 | 0.543543 | 0.917750 | 0.916868 |
| 600 | 0.007600 | 0.604909 | 0.921250 | 0.921414 |
| 800 | 0.008500 | 0.588405 | 0.922750 | 0.922533 |
| 1000 | 0.007300 | 0.618596 | 0.925500 | 0.925138 |
| 1200 | 0.006300 | 0.628956 | 0.923250 | 0.922644 |
2025.02.17 기준
- roc_auc_socre = 0.94
Dacon순위 24위
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- Languages
- en
- Created
- Feb 12, 2025
- Updated
- Feb 17, 2025