Hugging Face's logo

helling100
/
Regression_bert_1

Text Classification
Transformers
TensorFlow
distilbert
generated_from_keras_callback
Inference Endpoints
Model card Files Community
helling100 commited on
Commit
249b497
•
1 Parent(s): 3fc718a

Upload TFDistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +20 -22
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,16 +14,14 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.2128
18
- - Train Mae: 0.2623
19
- - Train Mse: 0.1098
20
- - Train Accuracy: 0.8615
21
- - Train R2-score: 0.8081
22
- - Validation Loss: 0.1657
23
- - Validation Mae: 0.3472
24
- - Validation Mse: 0.1644
25
- - Validation Accuracy: 0.7027
26
- - Validation R2-score: 0.8599
27
  - Epoch: 9
28
 
29
  ## Model description
@@ -48,18 +46,18 @@ The following hyperparameters were used during training:
48
 
49
  ### Training results
50
 
51
- | Train Loss | Train Mae | Train Mse | Train Accuracy | Train R2-score | Validation Loss | Validation Mae | Validation Mse | Validation Accuracy | Validation R2-score | Epoch |
52
- |:----------:|:---------:|:---------:|:--------------:|:--------------:|:---------------:|:--------------:|:--------------:|:-------------------:|:-------------------:|:-----:|
53
- | 0.6256 | 0.3353 | 0.1579 | 0.7615 | 0.4024 | 0.2916 | 0.4907 | 0.2909 | 0.3243 | 0.7810 | 0 |
54
- | 0.3639 | 0.3290 | 0.1605 | 0.7077 | 0.3874 | 0.3009 | 0.5004 | 0.3003 | 0.3243 | 0.7733 | 1 |
55
- | 0.1835 | 0.2940 | 0.1415 | 0.6615 | 0.7274 | 0.2086 | 0.3992 | 0.2075 | 0.3243 | 0.8417 | 2 |
56
- | 0.1707 | 0.2955 | 0.1462 | 0.5846 | 0.7594 | 0.1872 | 0.3705 | 0.1859 | 0.3243 | 0.8547 | 3 |
57
- | 0.1628 | 0.2740 | 0.1251 | 0.8077 | 0.7588 | 0.1867 | 0.3707 | 0.1854 | 0.4595 | 0.8547 | 4 |
58
- | 0.1541 | 0.2695 | 0.1221 | 0.7769 | 0.7405 | 0.1851 | 0.3696 | 0.1839 | 0.5946 | 0.8549 | 5 |
59
- | 0.2239 | 0.2983 | 0.1388 | 0.7154 | 0.7428 | 0.2561 | 0.4564 | 0.2552 | 0.3243 | 0.7987 | 6 |
60
- | 0.1998 | 0.2815 | 0.1295 | 0.7538 | 0.7537 | 0.1979 | 0.3872 | 0.1968 | 0.3514 | 0.8473 | 7 |
61
- | 0.1682 | 0.2743 | 0.1260 | 0.7692 | 0.7532 | 0.1515 | 0.3350 | 0.1500 | 0.9730 | 0.8691 | 8 |
62
- | 0.2128 | 0.2623 | 0.1098 | 0.8615 | 0.8081 | 0.1657 | 0.3472 | 0.1644 | 0.7027 | 0.8599 | 9 |
63
 
64
 
65
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.1363
18
+ - Train Mae: 0.2288
19
+ - Train Mse: 0.0928
20
+ - Train R2-score: 0.8065
21
+ - Validation Loss: 0.0994
22
+ - Validation Mae: 0.2297
23
+ - Validation Mse: 0.0947
24
+ - Validation R2-score: 0.8577
 
 
25
  - Epoch: 9
26
 
27
  ## Model description
 
46
 
47
  ### Training results
48
 
49
+ | Train Loss | Train Mae | Train Mse | Train R2-score | Validation Loss | Validation Mae | Validation Mse | Validation R2-score | Epoch |
50
+ |:----------:|:---------:|:---------:|:--------------:|:---------------:|:--------------:|:--------------:|:-------------------:|:-----:|
51
+ | 0.4890 | 0.3421 | 0.1734 | 0.1995 | 0.1833 | 0.3662 | 0.1820 | 0.8574 | 0 |
52
+ | 0.1771 | 0.3049 | 0.1526 | 0.7380 | 0.1715 | 0.3553 | 0.1701 | 0.8642 | 1 |
53
+ | 0.1563 | 0.2820 | 0.1337 | 0.7278 | 0.1823 | 0.3662 | 0.1810 | 0.8577 | 2 |
54
+ | 0.1578 | 0.2739 | 0.1288 | 0.7954 | 0.1689 | 0.3535 | 0.1674 | 0.8641 | 3 |
55
+ | 0.2301 | 0.3034 | 0.1383 | 0.7186 | 0.1158 | 0.3132 | 0.1136 | 0.8866 | 4 |
56
+ | 0.2067 | 0.2768 | 0.1203 | 0.6568 | 0.1462 | 0.3375 | 0.1445 | 0.8772 | 5 |
57
+ | 0.1631 | 0.2726 | 0.1200 | 0.7418 | 0.1646 | 0.3473 | 0.1632 | 0.8659 | 6 |
58
+ | 0.1689 | 0.2843 | 0.1269 | 0.4797 | 0.0993 | 0.2797 | 0.0972 | 0.8961 | 7 |
59
+ | 0.1756 | 0.2115 | 0.0726 | 0.8657 | 0.1047 | 0.2537 | 0.1027 | 0.8923 | 8 |
60
+ | 0.1363 | 0.2288 | 0.0928 | 0.8065 | 0.0994 | 0.2297 | 0.0947 | 0.8577 | 9 |
61
 
62
 
63
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ab99698bfd9da1b0470cd06975d65b41f4105fd3fddd1f2fac0ed8d82bbc31dc
3
  size 267952072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54eed3b23c20c10ab725b49aad1a24a26752d4921025720c0d44c025d301ac2c
3
  size 267952072