BERT-crime-analysis / config.json
Luna-Skywalker's picture
BERT-crime-analysis
b9a9ae1 verified
raw
history blame contribute delete
No virus
2.65 kB
{
"_name_or_path": "google-bert/bert-base-uncased",
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "ARSON",
"1": "ASSAULT",
"2": "BATTERY",
"3": "BURGLARY",
"4": "CONCEALED CARRY LICENSE VIOLATION",
"5": "CRIM SEXUAL ASSAULT",
"6": "CRIMINAL DAMAGE",
"7": "CRIMINAL TRESPASS",
"8": "DECEPTIVE PRACTICE",
"9": "DOMESTIC VIOLENCE",
"10": "GAMBLING",
"11": "HOMICIDE",
"12": "HUMAN TRAFFICKING",
"13": "INTERFERENCE WITH PUBLIC OFFICER",
"14": "INTIMIDATION",
"15": "KIDNAPPING",
"16": "LIQUOR LAW VIOLATION",
"17": "MOTOR VEHICLE THEFT",
"18": "NARCOTICS",
"19": "NON-CRIMINAL",
"20": "NON-CRIMINAL (SUBJECT SPECIFIED)",
"21": "OBSCENITY",
"22": "OFFENSE INVOLVING CHILDREN",
"23": "OTHER NARCOTIC VIOLATION",
"24": "OTHER OFFENSE",
"25": "PROSTITUTION",
"26": "PUBLIC INDECENCY",
"27": "PUBLIC PEACE VIOLATION",
"28": "RITUALISM",
"29": "ROBBERY",
"30": "SEX OFFENSE",
"31": "STALKING",
"32": "THEFT",
"33": "WEAPONS VIOLATION"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"ARSON": 0,
"ASSAULT": 1,
"BATTERY": 2,
"BURGLARY": 3,
"CONCEALED CARRY LICENSE VIOLATION": 4,
"CRIM SEXUAL ASSAULT": 5,
"CRIMINAL DAMAGE": 6,
"CRIMINAL TRESPASS": 7,
"DECEPTIVE PRACTICE": 8,
"DOMESTIC VIOLENCE": 9,
"GAMBLING": 10,
"HOMICIDE": 11,
"HUMAN TRAFFICKING": 12,
"INTERFERENCE WITH PUBLIC OFFICER": 13,
"INTIMIDATION": 14,
"KIDNAPPING": 15,
"LIQUOR LAW VIOLATION": 16,
"MOTOR VEHICLE THEFT": 17,
"NARCOTICS": 18,
"NON-CRIMINAL": 19,
"NON-CRIMINAL (SUBJECT SPECIFIED)": 20,
"OBSCENITY": 21,
"OFFENSE INVOLVING CHILDREN": 22,
"OTHER NARCOTIC VIOLATION": 23,
"OTHER OFFENSE": 24,
"PROSTITUTION": 25,
"PUBLIC INDECENCY": 26,
"PUBLIC PEACE VIOLATION": 27,
"RITUALISM": 28,
"ROBBERY": 29,
"SEX OFFENSE": 30,
"STALKING": 31,
"THEFT": 32,
"WEAPONS VIOLATION": 33
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.39.3",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 30522
}