Team Ai
Modelpublic

ExecuteAutomation/bert-base-text-classification-model

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes12downloads
config.json44 linesDownload Raw Back to root
1{2  "_name_or_path": "bert-base-uncased",3  "architectures": [4    "BertForSequenceClassification"5  ],6  "attention_probs_dropout_prob": 0.1,7  "classifier_dropout": null,8  "gradient_checkpointing": false,9  "hidden_act": "gelu",10  "hidden_dropout_prob": 0.1,11  "hidden_size": 768,12  "id2label": {13    "0": "sadness",14    "1": "joy",15    "2": "love",16    "3": "anger",17    "4": "fear",18    "5": "surprise"19  },20  "initializer_range": 0.02,21  "intermediate_size": 3072,22  "label2id": {23    "anger": 3,24    "fear": 4,25    "joy": 1,26    "love": 2,27    "sadness": 0,28    "surprise": 529  },30  "layer_norm_eps": 1e-12,31  "max_position_embeddings": 512,32  "model_type": "bert",33  "num_attention_heads": 12,34  "num_hidden_layers": 12,35  "pad_token_id": 0,36  "position_embedding_type": "absolute",37  "problem_type": "single_label_classification",38  "torch_dtype": "float32",39  "transformers_version": "4.48.0",40  "type_vocab_size": 2,41  "use_cache": true,42  "vocab_size": 3052243}44