ShihHsuanChen commited on
Commit
f8e9a2d
·
verified ·
1 Parent(s): 23beef3

Update Model card

Browse files
Files changed (1) hide show
  1. README.md +23 -6
README.md CHANGED
@@ -7,13 +7,30 @@ base_model:
7
  - microsoft/deberta-v3-base
8
  library_name: transformers
9
  ---
10
- - Training Notebook:
11
- [https://www.kaggle.com/code/shihhsuanchen/emotion-detection-train]
 
 
12
 
13
- - Evaluation Notebook:
14
- [https://www.kaggle.com/code/shihhsuanchen/emotion-detection-eval]
15
-
16
- - Evaluation Results
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  | | sadness | joy | love | anger | fear | surprise | Total (Macro) |
18
  |-----------|----------|----------|----------|----------|----------|----------|---------------|
19
  | Accuracy | 0.985420 | 0.965309 | 0.974359 | 0.983912 | 0.980392 | 0.985923 | 0.979219 |
 
7
  - microsoft/deberta-v3-base
8
  library_name: transformers
9
  ---
10
+ # Training
11
+ - Code: [Training Notebook](https://www.kaggle.com/code/shihhsuanchen/emotion-detection-train)
12
+ - Criteria: Best validation loss
13
+ - Training configuration:
14
 
15
+ ```json
16
+ {
17
+ "seed": 567,
18
+ "ddp": true,
19
+ "learning_rate": 5e-05,
20
+ "train_batch_size": 80,
21
+ "valid_batch_size": 80,
22
+ "lr_scheduler_type": "linear",
23
+ "num_epochs": 20,
24
+ "num_warmup_steps": 125,
25
+ "max_train_steps": null,
26
+ "max_valid_steps": null,
27
+ "max_length": 72
28
+ }
29
+ ```
30
+
31
+ # Evaluation
32
+ - Code: [Evaluation Notebook](https://www.kaggle.com/code/shihhsuanchen/emotion-detection-eval)
33
+ - Results
34
  | | sadness | joy | love | anger | fear | surprise | Total (Macro) |
35
  |-----------|----------|----------|----------|----------|----------|----------|---------------|
36
  | Accuracy | 0.985420 | 0.965309 | 0.974359 | 0.983912 | 0.980392 | 0.985923 | 0.979219 |