config.json
1.9 KB · 89 lines · json Raw
1 {
2 "_name_or_path": "roberta-base",
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 "id2label": {
14 "0": "admiration",
15 "1": "amusement",
16 "2": "anger",
17 "3": "annoyance",
18 "4": "approval",
19 "5": "caring",
20 "6": "confusion",
21 "7": "curiosity",
22 "8": "desire",
23 "9": "disappointment",
24 "10": "disapproval",
25 "11": "disgust",
26 "12": "embarrassment",
27 "13": "excitement",
28 "14": "fear",
29 "15": "gratitude",
30 "16": "grief",
31 "17": "joy",
32 "18": "love",
33 "19": "nervousness",
34 "20": "optimism",
35 "21": "pride",
36 "22": "realization",
37 "23": "relief",
38 "24": "remorse",
39 "25": "sadness",
40 "26": "surprise",
41 "27": "neutral"
42 },
43 "initializer_range": 0.02,
44 "intermediate_size": 3072,
45 "label2id": {
46 "admiration": 0,
47 "amusement": 1,
48 "anger": 2,
49 "annoyance": 3,
50 "approval": 4,
51 "caring": 5,
52 "confusion": 6,
53 "curiosity": 7,
54 "desire": 8,
55 "disappointment": 9,
56 "disapproval": 10,
57 "disgust": 11,
58 "embarrassment": 12,
59 "excitement": 13,
60 "fear": 14,
61 "gratitude": 15,
62 "grief": 16,
63 "joy": 17,
64 "love": 18,
65 "nervousness": 19,
66 "neutral": 27,
67 "optimism": 20,
68 "pride": 21,
69 "realization": 22,
70 "relief": 23,
71 "remorse": 24,
72 "sadness": 25,
73 "surprise": 26
74 },
75 "layer_norm_eps": 1e-05,
76 "max_position_embeddings": 514,
77 "model_type": "roberta",
78 "num_attention_heads": 12,
79 "num_hidden_layers": 12,
80 "pad_token_id": 1,
81 "position_embedding_type": "absolute",
82 "problem_type": "multi_label_classification",
83 "torch_dtype": "float32",
84 "transformers_version": "4.21.3",
85 "type_vocab_size": 1,
86 "use_cache": true,
87 "vocab_size": 50265
88 }
89