Hugging Face's logo

cruiser
/
distilbert_model_kaggle_200_epoch

Text Classification
Transformers
TensorFlow
distilbert
generated_from_keras_callback
text-embeddings-inference
Model card Files
xet
Community
cruiser commited on
Commit
c0bcae2
·
1 Parent(s): 466f294

Training in progress epoch 0

Browse files
Files changed (3) hide show
  1. README.md +4 -8
  2. config.json +6 -6
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,9 +14,9 @@ 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: 1.0986
18
- - Train Accuracy: 0.3451
19
- - Epoch: 4
20
 
21
  ## Model description
22
 
@@ -42,11 +42,7 @@ The following hyperparameters were used during training:
42
 
43
  | Train Loss | Train Accuracy | Epoch |
44
  |:----------:|:--------------:|:-----:|
45
- | 1.1115 | 0.3679 | 0 |
46
- | 1.0986 | 0.3445 | 1 |
47
- | 1.0986 | 0.3445 | 2 |
48
- | 1.0986 | 0.3423 | 3 |
49
- | 1.0986 | 0.3451 | 4 |
50
 
51
 
52
  ### 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: 1.1017
18
+ - Train Accuracy: 0.3545
19
+ - Epoch: 0
20
 
21
  ## Model description
22
 
 
42
 
43
  | Train Loss | Train Accuracy | Epoch |
44
  |:----------:|:--------------:|:-----:|
45
+ | 1.1017 | 0.3545 | 0 |
 
 
 
 
46
 
47
 
48
  ### Framework versions
config.json CHANGED
@@ -9,15 +9,15 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2"
15
  },
16
  "initializer_range": 0.02,
17
  "label2id": {
18
- "LABEL_0": 0,
19
- "LABEL_1": 1,
20
- "LABEL_2": 2
21
  },
22
  "max_position_embeddings": 512,
23
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "Negative",
13
+ "1": "Neutral",
14
+ "2": "Positive"
15
  },
16
  "initializer_range": 0.02,
17
  "label2id": {
18
+ "Negative": 0,
19
+ "Neutral": 1,
20
+ "Positive": 2
21
  },
22
  "max_position_embeddings": 512,
23
  "model_type": "distilbert",
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d3b0f6cb4f45644bd7f2c8599c915ca72f671657feac4381d9140c3ca8cf3daa
3
  size 267958216
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc1955f6679608a7f44c2524550253d4229b979169bd06d9e1b427963a0f10ac
3
  size 267958216