gaaahee commited on
Commit
8c692fc
·
verified ·
1 Parent(s): 72d0e30

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +37 -0
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "bert",
3
+ "base_model": "monologg/kobert",
4
+ "num_labels": 3,
5
+ "labels": {
6
+ "0": "옹호",
7
+ "1": "중립",
8
+ "2": "비판"
9
+ },
10
+ "max_length": 256,
11
+ "dropout": 0.3,
12
+ "metrics": {
13
+ "test_accuracy": 0.7336561743341404,
14
+ "test_f1_macro": 0.7335882497408175,
15
+ "best_val_f1": 0.7595240086537992
16
+ },
17
+ "training_config": {
18
+ "model_name": "monologg/kobert",
19
+ "max_length": 256,
20
+ "dropout": 0.3,
21
+ "batch_size": 64,
22
+ "epochs": 10,
23
+ "learning_rate": 2e-05,
24
+ "weight_decay": 0.01,
25
+ "max_grad_norm": 1.0,
26
+ "use_focal_loss": true,
27
+ "focal_gamma": 2.0,
28
+ "early_stop_patience": 3,
29
+ "early_stop_min_delta": 0.001,
30
+ "val_size": 0.1,
31
+ "test_size": 0.1,
32
+ "wandb_project": "kobert-stance-16k",
33
+ "wandb_run_name": null,
34
+ "save_checkpoint_every": 1,
35
+ "checkpoint_dir": "/content/checkpoints"
36
+ }
37
+ }