| { | |
| "architectures": ["BertForMaskedLM"], | |
| "attention_probs_dropout_prob": 0.1, | |
| "bos_token_id": 0, | |
| "directionality": "bidi", | |
| "eos_token_id": 2, | |
| "hidden_act": "gelu", | |
| "hidden_dropout_prob": 0.1, | |
| "hidden_size": 1024, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 4096, | |
| "layer_norm_eps": 1e-12, | |
| "max_position_embeddings": 512, | |
| "model_type": "bert", | |
| "num_attention_heads": 16, | |
| "num_hidden_layers": 24, | |
| "output_past": true, | |
| "pad_token_id": 0, | |
| "pooler_fc_size": 768, | |
| "pooler_num_attention_heads": 12, | |
| "pooler_num_fc_layers": 3, | |
| "pooler_size_per_head": 128, | |
| "pooler_type": "first_token_transform", | |
| "type_vocab_size": 2, | |
| "vocab_size": 21128, | |
| "id2label": { | |
| "0": "自杀未遂", | |
| "1": "自杀准备行为", | |
| "2": "自杀计划", | |
| "3": "主动自杀意图", | |
| "4": "被动自杀意图", | |
| "5": "用户攻击行为", | |
| "6": "他人攻击行为", | |
| "7": "自伤行为", | |
| "8": "自伤意图", | |
| "9": "关于自杀的探索", | |
| "10": "与自杀/自伤/攻击行为无关" | |
| }, | |
| "label2id": { | |
| "自杀未遂": 0, | |
| "自杀准备行为": 1, | |
| "自杀计划": 2, | |
| "主动自杀意图": 3, | |
| "被动自杀意图": 4, | |
| "用户攻击行为": 5, | |
| "他人攻击行为": 6, | |
| "自伤行为": 7, | |
| "自伤意图": 8, | |
| "关于自杀的探索": 9, | |
| "与自杀/自伤/攻击行为无关": 10 | |
| } | |
| } | |