Hugging Face's logo

helling100
/
Regression_albert_5

Text Classification
Transformers
TensorFlow
distilbert
generated_from_keras_callback
Model card Files
xet
Community
helling100 commited on
Commit
8f64a26
·
1 Parent(s): 5e02a78

Upload TFDistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +22 -20
  2. tf_model.h5 +2 -2
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.1814
18
- - Train Mae: 0.2938
19
- - Train Mse: 0.1429
20
- - Train Accuracy: 0.6500
21
- - Validation Loss: 0.1964
22
- - Validation Mae: 0.3819
23
- - Validation Mse: 0.1952
24
- - Validation Accuracy: 0.5811
 
 
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 Accuracy | Validation Loss | Validation Mae | Validation Mse | Validation Accuracy | Epoch |
50
- |:----------:|:---------:|:---------:|:--------------:|:---------------:|:--------------:|:--------------:|:-------------------:|:-----:|
51
- | 1.1449 | 0.5862 | 0.4388 | 0.5115 | 0.3923 | 0.5808 | 0.3920 | 0.3243 | 0 |
52
- | 0.3044 | 0.3674 | 0.2078 | 0.5654 | 0.2169 | 0.4102 | 0.2158 | 0.3243 | 1 |
53
- | 0.1775 | 0.2928 | 0.1414 | 0.6923 | 0.1833 | 0.3691 | 0.1819 | 0.6351 | 2 |
54
- | 0.1832 | 0.2879 | 0.1347 | 0.7269 | 0.1846 | 0.3672 | 0.1833 | 0.6486 | 3 |
55
- | 0.1642 | 0.2894 | 0.1411 | 0.7038 | 0.1930 | 0.3802 | 0.1917 | 0.3919 | 4 |
56
- | 0.1657 | 0.2888 | 0.1386 | 0.6615 | 0.1856 | 0.3711 | 0.1843 | 0.6081 | 5 |
57
- | 0.1787 | 0.2922 | 0.1425 | 0.6731 | 0.1960 | 0.3854 | 0.1947 | 0.4865 | 6 |
58
- | 0.1896 | 0.2842 | 0.1335 | 0.7231 | 0.1788 | 0.3646 | 0.1774 | 0.6351 | 7 |
59
- | 0.1712 | 0.2949 | 0.1465 | 0.6308 | 0.1876 | 0.3728 | 0.1863 | 0.5811 | 8 |
60
- | 0.1814 | 0.2938 | 0.1429 | 0.6500 | 0.1964 | 0.3819 | 0.1952 | 0.5811 | 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.1772
18
+ - Train Mae: 0.2939
19
+ - Train Mse: 0.1408
20
+ - Train R2-score: 0.5387
21
+ - Train Accuracy: 0.6538
22
+ - Validation Loss: 0.1822
23
+ - Validation Mae: 0.3656
24
+ - Validation Mse: 0.1809
25
+ - Validation R2-score: 0.7154
26
+ - Validation Accuracy: 0.7027
27
  - Epoch: 9
28
 
29
  ## Model description
 
48
 
49
  ### Training results
50
 
51
+ | Train Loss | Train Mae | Train Mse | Train R2-score | Train Accuracy | Validation Loss | Validation Mae | Validation Mse | Validation R2-score | Validation Accuracy | Epoch |
52
+ |:----------:|:---------:|:---------:|:--------------:|:--------------:|:---------------:|:--------------:|:--------------:|:-------------------:|:-------------------:|:-----:|
53
+ | 1.5062 | 0.5873 | 0.4379 | -0.5882 | 0.4615 | 0.4702 | 0.6369 | 0.4700 | 0.2677 | 0.3243 | 0 |
54
+ | 0.5726 | 0.4513 | 0.2799 | -0.2102 | 0.5538 | 0.2739 | 0.4733 | 0.2732 | 0.5908 | 0.3243 | 1 |
55
+ | 0.1934 | 0.3138 | 0.1581 | 0.3070 | 0.6154 | 0.1815 | 0.3648 | 0.1802 | 0.7161 | 0.6351 | 2 |
56
+ | 0.1820 | 0.2921 | 0.1420 | 0.5935 | 0.6731 | 0.1927 | 0.3805 | 0.1914 | 0.7023 | 0.6081 | 3 |
57
+ | 0.1631 | 0.2892 | 0.1408 | 0.3709 | 0.6538 | 0.1837 | 0.3715 | 0.1824 | 0.7124 | 0.6216 | 4 |
58
+ | 0.1821 | 0.2952 | 0.1445 | 0.3606 | 0.6615 | 0.1854 | 0.3706 | 0.1841 | 0.7113 | 0.5946 | 5 |
59
+ | 0.1691 | 0.2740 | 0.1261 | 0.5228 | 0.7692 | 0.1748 | 0.3595 | 0.1734 | 0.7235 | 0.8514 | 6 |
60
+ | 0.1525 | 0.2761 | 0.1273 | 0.4328 | 0.7577 | 0.1754 | 0.3607 | 0.1740 | 0.7227 | 0.8378 | 7 |
61
+ | 0.1912 | 0.2910 | 0.1370 | 0.5786 | 0.7000 | 0.2063 | 0.3964 | 0.2052 | 0.6847 | 0.4865 | 8 |
62
+ | 0.1772 | 0.2939 | 0.1408 | 0.5387 | 0.6538 | 0.1822 | 0.3656 | 0.1809 | 0.7154 | 0.7027 | 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:0a6b2d9c094263fae89f3c8b904c17241e097a2ccb2dffa69197276e0229d973
3
- size 267951808
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0fa9ae3ae37c376ed81b20ff67bf9729d8b16c84dfc16fe51ddbf4edb53f64a
3
+ size 267955144