Porameht commited on
Commit
fe0063a
·
verified ·
1 Parent(s): 9e6cece

Training in progress, epoch 1

Browse files
Files changed (3) hide show
  1. README.md +14 -14
  2. config.json +20 -20
  3. model.safetensors +1 -1
README.md CHANGED
@@ -1,31 +1,31 @@
1
  ---
2
- base_model: Geotrend/bert-base-th-cased
3
  library_name: transformers
4
  license: apache-2.0
 
 
 
5
  metrics:
6
  - accuracy
7
  - f1
8
  - precision
9
  - recall
10
- tags:
11
- - generated_from_trainer
12
  model-index:
13
- - name: bert-base-th-cased-intent-booking
14
  results: []
15
  ---
16
 
17
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
  should probably proofread and complete it, then remove this comment. -->
19
 
20
- # bert-base-th-cased-intent-booking
21
 
22
- This model is a fine-tuned version of [Geotrend/bert-base-th-cased](https://huggingface.co/Geotrend/bert-base-th-cased) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.2346
25
- - Accuracy: 0.9369
26
- - F1: 0.9340
27
- - Precision: 0.9471
28
- - Recall: 0.9369
29
 
30
  ## Model description
31
 
@@ -57,9 +57,9 @@ The following hyperparameters were used during training:
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
60
- | 1.9427 | 1.0 | 65 | 0.7129 | 0.8964 | 0.8947 | 0.9055 | 0.8964 |
61
- | 0.48 | 2.0 | 130 | 0.2325 | 0.9369 | 0.9365 | 0.9390 | 0.9369 |
62
- | 0.1783 | 3.0 | 195 | 0.2001 | 0.9505 | 0.9511 | 0.9578 | 0.9505 |
63
 
64
 
65
  ### Framework versions
 
1
  ---
 
2
  library_name: transformers
3
  license: apache-2.0
4
+ base_model: google-bert/bert-base-multilingual-cased
5
+ tags:
6
+ - generated_from_trainer
7
  metrics:
8
  - accuracy
9
  - f1
10
  - precision
11
  - recall
 
 
12
  model-index:
13
+ - name: bert-base-multilingual-cased-intent-booking
14
  results: []
15
  ---
16
 
17
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
  should probably proofread and complete it, then remove this comment. -->
19
 
20
+ # bert-base-multilingual-cased-intent-booking
21
 
22
+ This model is a fine-tuned version of [google-bert/bert-base-multilingual-cased](https://huggingface.co/google-bert/bert-base-multilingual-cased) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.3045
25
+ - Accuracy: 0.9189
26
+ - F1: 0.9155
27
+ - Precision: 0.9322
28
+ - Recall: 0.9189
29
 
30
  ## Model description
31
 
 
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
60
+ | 2.246 | 1.0 | 65 | 1.7430 | 0.4730 | 0.3680 | 0.3529 | 0.4730 |
61
+ | 0.972 | 2.0 | 130 | 0.3620 | 0.9369 | 0.9371 | 0.9417 | 0.9369 |
62
+ | 0.3069 | 3.0 | 195 | 0.2379 | 0.9414 | 0.9412 | 0.9490 | 0.9414 |
63
 
64
 
65
  ### Framework versions
config.json CHANGED
@@ -11,30 +11,30 @@
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
- "0": "LABEL_0",
15
- "1": "LABEL_1",
16
- "2": "LABEL_2",
17
- "3": "LABEL_3",
18
- "4": "LABEL_4",
19
- "5": "LABEL_5",
20
- "6": "LABEL_6",
21
- "7": "LABEL_7",
22
- "8": "LABEL_8",
23
- "9": "LABEL_9"
24
  },
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
  "label2id": {
28
- "LABEL_0": 0,
29
- "LABEL_1": 1,
30
- "LABEL_2": 2,
31
- "LABEL_3": 3,
32
- "LABEL_4": 4,
33
- "LABEL_5": 5,
34
- "LABEL_6": 6,
35
- "LABEL_7": 7,
36
- "LABEL_8": 8,
37
- "LABEL_9": 9
38
  },
39
  "layer_norm_eps": 1e-12,
40
  "max_position_embeddings": 512,
 
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
+ "0": "Inquire Facilities",
15
+ "1": "Ask for Information",
16
+ "2": "Modify Reservation",
17
+ "3": "Request Special Service",
18
+ "4": "Cancel Reservation",
19
+ "5": "Check Availability",
20
+ "6": "Make Reservation",
21
+ "7": "Suggest Attractions",
22
+ "8": "Greeting",
23
+ "9": "Verify Reservation"
24
  },
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
  "label2id": {
28
+ "Ask for Information": 1,
29
+ "Cancel Reservation": 4,
30
+ "Check Availability": 5,
31
+ "Greeting": 8,
32
+ "Inquire Facilities": 0,
33
+ "Make Reservation": 6,
34
+ "Modify Reservation": 2,
35
+ "Request Special Service": 3,
36
+ "Suggest Attractions": 7,
37
+ "Verify Reservation": 9
38
  },
39
  "layer_norm_eps": 1e-12,
40
  "max_position_embeddings": 512,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f0711b8163c2f4c6e22b11acd76e4691408854e4c564f2ea478cb451803f483b
3
  size 370350032
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1128a4928582624b4321680d5cb1f2e7780e9f1f87aea24e407a47231daac849
3
  size 370350032