martincc98 commited on
Commit
7b35f29
1 Parent(s): 4f9ecef

Upload config

Browse files
Files changed (1) hide show
  1. config.json +60 -60
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "bert_baseline_model_train_a",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
@@ -9,67 +9,67 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3",
16
- "4": "LABEL_4",
17
- "5": "LABEL_5",
18
- "6": "LABEL_6",
19
- "7": "LABEL_7",
20
- "8": "LABEL_8",
21
- "9": "LABEL_9",
22
- "10": "LABEL_10",
23
- "11": "LABEL_11",
24
- "12": "LABEL_12",
25
- "13": "LABEL_13",
26
- "14": "LABEL_14",
27
- "15": "LABEL_15",
28
- "16": "LABEL_16",
29
- "17": "LABEL_17",
30
- "18": "LABEL_18",
31
- "19": "LABEL_19",
32
- "20": "LABEL_20",
33
- "21": "LABEL_21",
34
- "22": "LABEL_22",
35
- "23": "LABEL_23",
36
- "24": "LABEL_24",
37
- "25": "LABEL_25",
38
- "26": "LABEL_26",
39
- "27": "LABEL_27",
40
- "28": "LABEL_28"
41
  },
42
  "initializer_range": 0.02,
43
  "label2id": {
44
- "LABEL_0": 0,
45
- "LABEL_1": 1,
46
- "LABEL_10": 10,
47
- "LABEL_11": 11,
48
- "LABEL_12": 12,
49
- "LABEL_13": 13,
50
- "LABEL_14": 14,
51
- "LABEL_15": 15,
52
- "LABEL_16": 16,
53
- "LABEL_17": 17,
54
- "LABEL_18": 18,
55
- "LABEL_19": 19,
56
- "LABEL_2": 2,
57
- "LABEL_20": 20,
58
- "LABEL_21": 21,
59
- "LABEL_22": 22,
60
- "LABEL_23": 23,
61
- "LABEL_24": 24,
62
- "LABEL_25": 25,
63
- "LABEL_26": 26,
64
- "LABEL_27": 27,
65
- "LABEL_28": 28,
66
- "LABEL_3": 3,
67
- "LABEL_4": 4,
68
- "LABEL_5": 5,
69
- "LABEL_6": 6,
70
- "LABEL_7": 7,
71
- "LABEL_8": 8,
72
- "LABEL_9": 9
73
  },
74
  "max_position_embeddings": 512,
75
  "model_type": "distilbert",
@@ -81,6 +81,6 @@
81
  "sinusoidal_pos_embds": false,
82
  "tie_weights_": true,
83
  "torch_dtype": "float32",
84
- "transformers_version": "4.39.0",
85
  "vocab_size": 30522
86
  }
 
1
  {
2
+ "_name_or_path": "martincc98/srl_bert",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "ARG0",
13
+ "1": "ARG1",
14
+ "2": "ARG1-DSP",
15
+ "3": "ARG2",
16
+ "4": "ARG3",
17
+ "5": "ARG4",
18
+ "6": "ARG5",
19
+ "7": "ARGA",
20
+ "8": "ARGM-ADJ",
21
+ "9": "ARGM-ADV",
22
+ "10": "ARGM-CAU",
23
+ "11": "ARGM-COM",
24
+ "12": "ARGM-CXN",
25
+ "13": "ARGM-DIR",
26
+ "14": "ARGM-DIS",
27
+ "15": "ARGM-EXT",
28
+ "16": "ARGM-GOL",
29
+ "17": "ARGM-LOC",
30
+ "18": "ARGM-LVB",
31
+ "19": "ARGM-MNR",
32
+ "20": "ARGM-MOD",
33
+ "21": "ARGM-NEG",
34
+ "22": "ARGM-PRD",
35
+ "23": "ARGM-PRP",
36
+ "24": "ARGM-PRR",
37
+ "25": "ARGM-REC",
38
+ "26": "ARGM-TMP",
39
+ "27": "O",
40
+ "28": "V"
41
  },
42
  "initializer_range": 0.02,
43
  "label2id": {
44
+ "ARG0": 0,
45
+ "ARG1": 1,
46
+ "ARG1-DSP": 2,
47
+ "ARG2": 3,
48
+ "ARG3": 4,
49
+ "ARG4": 5,
50
+ "ARG5": 6,
51
+ "ARGA": 7,
52
+ "ARGM-ADJ": 8,
53
+ "ARGM-ADV": 9,
54
+ "ARGM-CAU": 10,
55
+ "ARGM-COM": 11,
56
+ "ARGM-CXN": 12,
57
+ "ARGM-DIR": 13,
58
+ "ARGM-DIS": 14,
59
+ "ARGM-EXT": 15,
60
+ "ARGM-GOL": 16,
61
+ "ARGM-LOC": 17,
62
+ "ARGM-LVB": 18,
63
+ "ARGM-MNR": 19,
64
+ "ARGM-MOD": 20,
65
+ "ARGM-NEG": 21,
66
+ "ARGM-PRD": 22,
67
+ "ARGM-PRP": 23,
68
+ "ARGM-PRR": 24,
69
+ "ARGM-REC": 25,
70
+ "ARGM-TMP": 26,
71
+ "O": 27,
72
+ "V": 28
73
  },
74
  "max_position_embeddings": 512,
75
  "model_type": "distilbert",
 
81
  "sinusoidal_pos_embds": false,
82
  "tie_weights_": true,
83
  "torch_dtype": "float32",
84
+ "transformers_version": "4.39.1",
85
  "vocab_size": 30522
86
  }