vazish commited on
Commit
dc8fa51
·
verified ·
1 Parent(s): 5add30f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -25
README.md CHANGED
@@ -16,7 +16,7 @@ datasets:
16
 
17
  ## Cross-Encoder for MS Marco with TinyBert
18
 
19
- This is a fine-tuned version of the model checkpointed at [cross-encoder/ms-marco-TinyBert-L-2](https://huggingface.co/cross-encoder/ms-marco-TinyBERT-L-2).
20
 
21
  It was fine-tuned on html tags and labels generated using [Fathom](https://mozilla.github.io/fathom/commands/label.html).
22
 
@@ -51,31 +51,26 @@ More information on how the model was trained can be found here: https://github.
51
  # Model Performance
52
  ```
53
  Test Performance:
54
- Precision: 0.9653
55
- Recall: 0.9648
56
- F1: 0.9644
57
 
58
- precision recall f1-score support
59
 
60
- CC Expiration 1.000 0.625 0.769 16
61
- CC Expiration Month 0.919 0.944 0.932 36
62
- CC Expiration Year 0.897 0.946 0.921 37
63
- CC Name 0.938 0.968 0.952 31
64
- CC Number 0.926 1.000 0.962 50
65
- CC Payment Type 0.903 0.867 0.884 75
66
- CC Security Code 0.975 0.951 0.963 41
67
- CC Type 0.917 0.786 0.846 14
68
- Confirm Password 0.911 0.895 0.903 57
69
- Email 0.933 0.959 0.946 73
70
- First Name 0.833 1.000 0.909 5
71
- Form 0.974 0.974 0.974 39
72
- Last Name 0.667 0.800 0.727 5
73
- New Password 0.929 0.938 0.933 97
74
- Other 0.985 0.985 0.985 1235
75
- Phone 1.000 0.667 0.800 3
76
- Zip Code 0.909 0.938 0.923 32
77
 
78
- accuracy 0.965 1846
79
- macro avg 0.919 0.897 0.902 1846
80
- weighted avg 0.965 0.965 0.964 1846
81
  ```
 
16
 
17
  ## Cross-Encoder for MS Marco with TinyBert
18
 
19
+ This is a fine-tuned version of the model checkpointed at [cross-encoder/ms-marco-TinyBert-L-2-v2](https://huggingface.co/cross-encoder/ms-marco-TinyBERT-L-2-v2).
20
 
21
  It was fine-tuned on html tags and labels generated using [Fathom](https://mozilla.github.io/fathom/commands/label.html).
22
 
 
51
  # Model Performance
52
  ```
53
  Test Performance:
54
+ Precision: 0.913
55
+ Recall: 0.872
56
+ F1: 0.887
57
 
58
+ precision recall f1-score support
59
 
60
+ cc-csc 0.943 0.950 0.946 139
61
+ cc-exp 1.000 0.883 0.938 60
62
+ cc-exp-month 0.954 0.922 0.938 90
63
+ cc-exp-year 0.904 0.934 0.919 91
64
+ cc-name 0.835 0.989 0.905 92
65
+ cc-number 0.953 0.970 0.961 167
66
+ cc-type 0.920 0.940 0.930 183
67
+ email 0.918 0.927 0.922 205
68
+ given-name 0.727 0.421 0.533 19
69
+ last-name 0.833 0.588 0.690 17
70
+ other 0.994 0.994 0.994 8000
71
+ postal-code 0.980 0.951 0.965 102
 
 
 
 
 
72
 
73
+ accuracy 0.985 9165
74
+ macro avg 0.913 0.872 0.887 9165
75
+ weighted avg 0.986 0.985 0.985 9165
76
  ```