CoolFace
Modelpublic

Rudhan/priority_classifier_output

sourceHugging Faceapache-2.0updated 5mo agoView on Hugging Face
0likes2downloads
config.json39 linesDownload Raw Back to root
1{2  "activation": "gelu",3  "architectures": [4    "DistilBertForSequenceClassification"5  ],6  "attention_dropout": 0.1,7  "bos_token_id": null,8  "dim": 768,9  "dropout": 0.1,10  "dtype": "float32",11  "eos_token_id": null,12  "hidden_dim": 3072,13  "id2label": {14    "0": "LABEL_0",15    "1": "LABEL_1",16    "2": "LABEL_2"17  },18  "initializer_range": 0.02,19  "label2id": {20    "LABEL_0": 0,21    "LABEL_1": 1,22    "LABEL_2": 223  },24  "max_position_embeddings": 512,25  "model_type": "distilbert",26  "n_heads": 12,27  "n_layers": 6,28  "pad_token_id": 0,29  "problem_type": "single_label_classification",30  "qa_dropout": 0.1,31  "seq_classif_dropout": 0.2,32  "sinusoidal_pos_embds": false,33  "tie_weights_": true,34  "tie_word_embeddings": true,35  "transformers_version": "5.0.0",36  "use_cache": false,37  "vocab_size": 3052238}39