{ "_name_or_path": "dslim/distilbert-NER", "activation": "gelu", "architectures": [ "DistilBertForTokenClassification" ], "attention_dropout": 0.1, "dim": 768, "dropout": 0.1, "hidden_dim": 3072, "id2label": { "0": "B-Actor", "1": "I-Actor", "2": "O", "3": "B-Plot", "4": "I-Plot", "5": "B-Opinion", "6": "I-Opinion", "7": "B-Award", "8": "I-Award", "9": "B-Year", "10": "B-Genre", "11": "B-Origin", "12": "I-Origin", "13": "B-Director", "14": "I-Director", "15": "I-Genre", "16": "I-Year", "17": "B-Soundtrack", "18": "I-Soundtrack", "19": "B-Relationship", "20": "I-Relationship", "21": "B-Character_Name", "22": "I-Character_Name", "23": "B-Quote", "24": "I-Quote" }, "initializer_range": 0.02, "label2id": { "B-Actor": 0, "B-Award": 7, "B-Character_Name": 21, "B-Director": 13, "B-Genre": 10, "B-Opinion": 5, "B-Origin": 11, "B-Plot": 3, "B-Quote": 23, "B-Relationship": 19, "B-Soundtrack": 17, "B-Year": 9, "I-Actor": 1, "I-Award": 8, "I-Character_Name": 22, "I-Director": 14, "I-Genre": 15, "I-Opinion": 6, "I-Origin": 12, "I-Plot": 4, "I-Quote": 24, "I-Relationship": 20, "I-Soundtrack": 18, "I-Year": 16, "O": 2 }, "max_position_embeddings": 512, "model_type": "distilbert", "n_heads": 12, "n_layers": 6, "output_past": true, "pad_token_id": 0, "qa_dropout": 0.1, "seq_classif_dropout": 0.2, "sinusoidal_pos_embds": false, "tie_weights_": true, "torch_dtype": "float32", "transformers_version": "4.41.1", "vocab_size": 28996 }