CoolFace
Modelpublic

mtebad/classification_model

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
1likes7downloads
config.json42 linesDownload Raw Back to root
1{2  "_name_or_path": "distilbert-base-uncased",3  "activation": "gelu",4  "architectures": [5    "DistilBertForSequenceClassification"6  ],7  "attention_dropout": 0.1,8  "dim": 768,9  "dropout": 0.1,10  "hidden_dim": 3072,11  "id2label": {12    "0": "anger",13    "1": "joy",14    "2": "love",15    "3": "sadness",16    "4": "surprise",17    "5": "fear"18  },19  "initializer_range": 0.02,20  "label2id": {21    "anger": 0,22    "fear": 5,23    "joy": 1,24    "love": 2,25    "sadness": 3,26    "surprise": 427  },28  "max_position_embeddings": 512,29  "model_type": "distilbert",30  "n_heads": 12,31  "n_layers": 6,32  "pad_token_id": 0,33  "problem_type": "single_label_classification",34  "qa_dropout": 0.1,35  "seq_classif_dropout": 0.2,36  "sinusoidal_pos_embds": false,37  "tie_weights_": true,38  "torch_dtype": "float32",39  "transformers_version": "4.30.2",40  "vocab_size": 3052241}42