Rocketknight1 HF staff commited on
Commit
6ccc74c
1 Parent(s): 28eb639
Files changed (3) hide show
  1. README.md +11 -3
  2. config.json +25 -13
  3. tf_model.h5 +2 -2
README.md CHANGED
@@ -12,9 +12,13 @@ probably proofread and complete it, then remove this comment. -->
12
 
13
  # europython-imdb
14
 
15
- This model is a fine-tuned version of [Rocketknight1/europython-imdb](https://huggingface.co/Rocketknight1/europython-imdb) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
-
 
 
 
 
18
 
19
  ## Model description
20
 
@@ -33,11 +37,15 @@ More information needed
33
  ### Training hyperparameters
34
 
35
  The following hyperparameters were used during training:
36
- - optimizer: None
37
  - training_precision: float32
38
 
39
  ### Training results
40
 
 
 
 
 
41
 
42
 
43
  ### Framework versions
 
12
 
13
  # europython-imdb
14
 
15
+ This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.1279
18
+ - Train Accuracy: 0.9548
19
+ - Validation Loss: 0.1595
20
+ - Validation Accuracy: 0.9418
21
+ - Epoch: 1
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - optimizer: {'name': 'Adam', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
41
  - training_precision: float32
42
 
43
  ### Training results
44
 
45
+ | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
46
+ |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
47
+ | 0.2073 | 0.9203 | 0.1486 | 0.9435 | 0 |
48
+ | 0.1279 | 0.9548 | 0.1595 | 0.9418 | 1 |
49
 
50
 
51
  ### Framework versions
config.json CHANGED
@@ -1,26 +1,38 @@
1
  {
2
- "_name_or_path": "Rocketknight1/europython-imdb",
3
  "architectures": [
4
- "RobertaForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
- "bos_token_id": 0,
8
- "classifier_dropout": null,
9
- "eos_token_id": 2,
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
 
 
 
 
13
  "initializer_range": 0.02,
14
  "intermediate_size": 3072,
15
- "layer_norm_eps": 1e-05,
16
- "max_position_embeddings": 514,
17
- "model_type": "roberta",
 
 
18
  "num_attention_heads": 12,
19
  "num_hidden_layers": 12,
20
- "pad_token_id": 1,
21
- "position_embedding_type": "absolute",
 
 
 
 
 
 
 
 
 
 
22
  "transformers_version": "4.21.0.dev0",
23
- "type_vocab_size": 1,
24
- "use_cache": true,
25
- "vocab_size": 50265
26
  }
 
1
  {
2
+ "_name_or_path": "microsoft/deberta-v3-base",
3
  "architectures": [
4
+ "DebertaV2ForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
 
 
 
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
  "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "Negative",
12
+ "1": "Positive"
13
+ },
14
  "initializer_range": 0.02,
15
  "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-07,
17
+ "max_position_embeddings": 512,
18
+ "max_relative_positions": -1,
19
+ "model_type": "deberta-v2",
20
+ "norm_rel_ebd": "layer_norm",
21
  "num_attention_heads": 12,
22
  "num_hidden_layers": 12,
23
+ "pad_token_id": 0,
24
+ "pooler_dropout": 0,
25
+ "pooler_hidden_act": "gelu",
26
+ "pooler_hidden_size": 768,
27
+ "pos_att_type": [
28
+ "p2c",
29
+ "c2p"
30
+ ],
31
+ "position_biased_input": false,
32
+ "position_buckets": 256,
33
+ "relative_attention": true,
34
+ "share_att_key": true,
35
  "transformers_version": "4.21.0.dev0",
36
+ "type_vocab_size": 0,
37
+ "vocab_size": 128100
 
38
  }
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:18fa14d632fc4c63bcb15d7bd151961eb20dfdcbee2422ae8c4702bf9b595d41
3
- size 498878336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:758dd785fc3710bc567f3a2748776b11ac07397c3e6605080b80d5679892343d
3
+ size 737986944