CoolFace
Modelpublic

rushikeshwalode/token_classification_NER

sourceHugging Faceapache-2.0updated 8d agoView on Hugging Face
1likes42downloads
config.json54 linesDownload Raw Back to root
1{2  "activation": "gelu",3  "architectures": [4    "DistilBertForTokenClassification"5  ],6  "attention_dropout": 0.1,7  "dim": 768,8  "dropout": 0.1,9  "hidden_dim": 3072,10  "id2label": {11    "0": "O",12    "1": "B-corporation",13    "2": "I-corporation",14    "3": "B-creative-work",15    "4": "I-creative-work",16    "5": "B-group",17    "6": "I-group",18    "7": "B-location",19    "8": "I-location",20    "9": "B-person",21    "10": "I-person",22    "11": "B-product",23    "12": "I-product"24  },25  "initializer_range": 0.02,26  "label2id": {27    "B-corporation": 1,28    "B-creative-work": 3,29    "B-group": 5,30    "B-location": 7,31    "B-person": 9,32    "B-product": 11,33    "I-corporation": 2,34    "I-creative-work": 4,35    "I-group": 6,36    "I-location": 8,37    "I-person": 10,38    "I-product": 12,39    "O": 040  },41  "max_position_embeddings": 512,42  "model_type": "distilbert",43  "n_heads": 12,44  "n_layers": 6,45  "pad_token_id": 0,46  "qa_dropout": 0.1,47  "seq_classif_dropout": 0.2,48  "sinusoidal_pos_embds": false,49  "tie_weights_": true,50  "torch_dtype": "float32",51  "transformers_version": "4.53.3",52  "vocab_size": 3052253}54