File size: 1,881 Bytes
296800d
d067132
296800d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
{
    "model_type": "emcoder",
    "auto_map": {
        "AutoConfig": "configuration_emcoder.EmCoderConfig",
        "AutoModel": "modeling_emcoder.EmCoder"
    },
    "architectures": [
        "EmCoder"
    ],
    "vocab_size": 50265,
    "max_seq_len": 512,
    "d_model": 768,
    "n_head": 12,
    "n_layers": 6,
    "d_ffn": 3072,
    "dropout": 0.15,
    "num_labels": 28,
    "id2label": {
        "0": "admiration",
        "1": "amusement",
        "2": "anger",
        "3": "annoyance",
        "4": "approval",
        "5": "caring",
        "6": "confusion",
        "7": "curiosity",
        "8": "desire",
        "9": "disappointment",
        "10": "disapproval",
        "11": "disgust",
        "12": "embarrassment",
        "13": "excitement",
        "14": "fear",
        "15": "gratitude",
        "16": "grief",
        "17": "joy",
        "18": "love",
        "19": "nervousness",
        "20": "optimism",
        "21": "pride",
        "22": "realization",
        "23": "relief",
        "24": "remorse",
        "25": "sadness",
        "26": "surprise",
        "27": "neutral"
    },
    "label2id": {
        "admiration": 0,
        "amusement": 1,
        "anger": 2,
        "annoyance": 3,
        "approval": 4,
        "caring": 5,
        "confusion": 6,
        "curiosity": 7,
        "desire": 8,
        "disappointment": 9,
        "disapproval": 10,
        "disgust": 11,
        "embarrassment": 12,
        "excitement": 13,
        "fear": 14,
        "gratitude": 15,
        "grief": 16,
        "joy": 17,
        "love": 18,
        "nervousness": 19,
        "optimism": 20,
        "pride": 21,
        "realization": 22,
        "relief": 23,
        "remorse": 24,
        "sadness": 25,
        "surprise": 26,
        "neutral": 27
    },
    "base_encoder_path": "models/v1/pretrain/checkpoints/epoch_2/step_40000"
}