CoolFace
Modelpublic

AliaeAI/dialogact_classification_bert_multilabel_v5

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes9downloads
config.json61 linesDownload Raw Back to root
1{2  "architectures": [3    "BertForSequenceClassification"4  ],5  "attention_probs_dropout_prob": 0.1,6  "classifier_dropout": null,7  "gradient_checkpointing": false,8  "hidden_act": "gelu",9  "hidden_dropout_prob": 0.1,10  "hidden_size": 768,11  "id2label": {12    "0": "thanks",13    "1": "question",14    "2": "bye",15    "3": "change-topic",16    "4": "confirm",17    "5": "confusion",18    "6": "doubt",19    "7": "empathy",20    "8": "greeting",21    "9": "inform",22    "10": "insulted",23    "11": "problem",24    "12": "refuse",25    "13": "success",26    "14": "wish"27  },28  "initializer_range": 0.02,29  "intermediate_size": 3072,30  "label2id": {31    "bye": 2,32    "change-topic": 3,33    "confirm": 4,34    "confusion": 5,35    "doubt": 6,36    "empathy": 7,37    "greeting": 8,38    "inform": 9,39    "insulted": 10,40    "problem": 11,41    "question": 1,42    "refuse": 12,43    "success": 13,44    "thanks": 0,45    "wish": 1446  },47  "layer_norm_eps": 1e-12,48  "max_position_embeddings": 512,49  "model_type": "bert",50  "num_attention_heads": 12,51  "num_hidden_layers": 12,52  "pad_token_id": 0,53  "position_embedding_type": "absolute",54  "problem_type": "multi_label_classification",55  "torch_dtype": "float32",56  "transformers_version": "4.55.0",57  "type_vocab_size": 2,58  "use_cache": true,59  "vocab_size": 3052260}61