Commit
·
04dc8c8
1
Parent(s):
31d96b9
Update README.md
Browse files
README.md
CHANGED
@@ -10,10 +10,48 @@ tags:
|
|
10 |
- spam detection
|
11 |
- email detection
|
12 |
- text classification
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
---
|
14 |
|
15 |
|
16 |
# Model Card for Text Classification for email-spam detection
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
```bibtex
|
19 |
@ModelCard{
|
|
|
10 |
- spam detection
|
11 |
- email detection
|
12 |
- text classification
|
13 |
+
inference: true
|
14 |
+
model-index:
|
15 |
+
- name: foduucom/Mail-spam-detection
|
16 |
+
results:
|
17 |
+
- task:
|
18 |
+
type: text-classification
|
19 |
+
metrics:
|
20 |
+
- type: precision
|
21 |
+
value: 0.866
|
22 |
---
|
23 |
|
24 |
|
25 |
# Model Card for Text Classification for email-spam detection
|
26 |
+
This model is based on Text classification using pytorch library. In this model we propose to used a torchtext library for tokenize & vectorize data.
|
27 |
+
It achieve the following results on the evalution set:
|
28 |
+
- accuracy : 0.866
|
29 |
+
|
30 |
+
## model architecture for text classification are below :
|
31 |
+
|
32 |
+
<div align="center">
|
33 |
+
<img width="640" alt="foduucom/Mail-spam-detection" src="https://huggingface.co/foduucom/thermal-image-object-detection/resolve/main/image.jpg">
|
34 |
+
</div>
|
35 |
+
|
36 |
+
|
37 |
+
### Training hyperparameters
|
38 |
+
|
39 |
+
The following hyperparameters were used during training:
|
40 |
+
- learning_rate: 3e-05
|
41 |
+
- train_batch_size: 1
|
42 |
+
- eval_batch_size: 1
|
43 |
+
- seed: 42
|
44 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
45 |
+
- lr_scheduler_type: linear
|
46 |
+
- lr_scheduler_warmup_steps: 500
|
47 |
+
- num_epochs: 1
|
48 |
+
|
49 |
+
|
50 |
+
|
51 |
+
### Framework versions
|
52 |
+
- Pytorch 1.10.0+cu111
|
53 |
+
- Tokenizers 0.10.3
|
54 |
+
|
55 |
|
56 |
```bibtex
|
57 |
@ModelCard{
|