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
853e446
·
1 Parent(s): ef6301c

Upload TFDistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +22 -20
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,14 +14,16 @@ 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.1783
18
- - Train Mae: 0.2454
19
- - Train Mse: 0.1031
20
- - Train R2-score: 0.4142
21
- - Validation Loss: 0.0857
22
- - Validation Mae: 0.2032
23
- - Validation Mse: 0.0821
24
- - Validation R2-score: 0.8682
 
 
25
  - Epoch: 9
26
 
27
  ## Model description
@@ -46,18 +48,18 @@ The following hyperparameters were used during training:
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.8164 | 0.3447 | 0.1992 | 0.5384 | 0.1862 | 0.3708 | 0.1849 | 0.8553 | 0 |
52
- | 0.2232 | 0.2734 | 0.1179 | 0.6082 | 0.2813 | 0.4837 | 0.2806 | 0.7862 | 1 |
53
- | 0.2817 | 0.3236 | 0.1602 | 0.6175 | 0.1569 | 0.3480 | 0.1553 | 0.8720 | 2 |
54
- | 0.2010 | 0.2856 | 0.1359 | 0.7224 | 0.2089 | 0.3973 | 0.2077 | 0.8394 | 3 |
55
- | 0.1825 | 0.2874 | 0.1365 | 0.7419 | 0.1608 | 0.3494 | 0.1593 | 0.8698 | 4 |
56
- | 0.1773 | 0.2616 | 0.1055 | 0.7859 | 0.2236 | 0.4185 | 0.2226 | 0.8301 | 5 |
57
- | 0.2332 | 0.3060 | 0.1529 | 0.7090 | 0.1610 | 0.3488 | 0.1595 | 0.8684 | 6 |
58
- | 0.1951 | 0.2734 | 0.1223 | 0.7305 | 0.1565 | 0.3434 | 0.1550 | 0.8695 | 7 |
59
- | 0.1249 | 0.2412 | 0.1018 | 0.8066 | 0.0774 | 0.2095 | 0.0756 | 0.8936 | 8 |
60
- | 0.1783 | 0.2454 | 0.1031 | 0.4142 | 0.0857 | 0.2032 | 0.0821 | 0.8682 | 9 |
61
 
62
 
63
  ### 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.1546
18
+ - Train Mae: 0.2782
19
+ - Train Mse: 0.1307
20
+ - Train Accuracy: 0.7385
21
+ - Train R2-score: 0.7406
22
+ - Validation Loss: 0.1848
23
+ - Validation Mae: 0.3685
24
+ - Validation Mse: 0.1836
25
+ - Validation Accuracy: 0.5135
26
+ - Validation R2-score: 0.8571
27
  - Epoch: 9
28
 
29
  ## Model description
 
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
+ | 1.0436 | 0.3757 | 0.2130 | 0.6769 | 0.5094 | 0.1461 | 0.3402 | 0.1443 | 0.9459 | 0.8772 | 0 |
54
+ | 0.2542 | 0.3126 | 0.1525 | 0.7154 | 0.6927 | 0.3952 | 0.5822 | 0.3950 | 0.3243 | 0.6797 | 1 |
55
+ | 0.1743 | 0.3049 | 0.1439 | 0.7231 | 0.6964 | 0.0748 | 0.2426 | 0.0722 | 0.9459 | 0.9034 | 2 |
56
+ | 0.4325 | 0.3154 | 0.1510 | 0.7077 | 0.6992 | 0.1663 | 0.3678 | 0.1641 | 0.8108 | 0.8491 | 3 |
57
+ | 0.2468 | 0.3069 | 0.1444 | 0.7385 | 0.3708 | 0.2735 | 0.4726 | 0.2728 | 0.3243 | 0.7953 | 4 |
58
+ | 0.3008 | 0.3204 | 0.1563 | 0.7308 | 0.7149 | 0.1543 | 0.3460 | 0.1527 | 0.9459 | 0.8730 | 5 |
59
+ | 0.1985 | 0.2822 | 0.1335 | 0.7000 | 0.7389 | 0.2090 | 0.3995 | 0.2079 | 0.3243 | 0.8411 | 6 |
60
+ | 0.2017 | 0.3021 | 0.1472 | 0.7231 | 0.7224 | 0.1305 | 0.3280 | 0.1286 | 0.9459 | 0.8820 | 7 |
61
+ | 0.2121 | 0.2961 | 0.1375 | 0.7308 | 0.7781 | 0.1706 | 0.3553 | 0.1692 | 0.9730 | 0.8655 | 8 |
62
+ | 0.1546 | 0.2782 | 0.1307 | 0.7385 | 0.7406 | 0.1848 | 0.3685 | 0.1836 | 0.5135 | 0.8571 | 9 |
63
 
64
 
65
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7542d52ad39641b25f36eb07b1f0cf0c2e3bde63a2294a0212c89dbe64f641c1
3
  size 267952072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a60cd720dcbcd04f2467227ed4fe26e013d1310041dcfa905bfb9067732fae79
3
  size 267952072