nehulagrawal commited on
Commit
04dc8c8
·
1 Parent(s): 31d96b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +38 -0
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{