fine-url-classifier / config.json
jonbarlow's picture
Upload DistilBertForSequenceClassification
d87c930 verified
raw
history blame
2.76 kB
{
"_name_or_path": "distilbert-base-uncased",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "Accident Reports",
"1": "Arrest Records",
"2": "Calls for Service",
"3": "Car GPS",
"4": "Citations",
"5": "Dispatch Logs",
"6": "Dispatch Recordings",
"7": "Field Contacts",
"8": "Incident Reports",
"9": "Misc Police Activity",
"10": "Officer Involved Shooting",
"11": "Stops",
"12": "Surveys",
"13": "Use of Force Reports",
"14": "Vehicle Pursuits",
"15": "Complaints & Misconduct",
"16": "Daily Activity Logs",
"17": "Training & Hiring Info",
"18": "Personnel Records",
"19": "Annual & Monthly Reports",
"20": "Budgets & Finances",
"21": "Contact Info & Agency Meta",
"22": "Geographic",
"23": "List of Data Sources",
"24": "Policies & Contracts",
"25": "Crime Maps & Reports",
"26": "Crime Statistics",
"27": "Media Bulletins",
"28": "Records Request Info",
"29": "Resources",
"30": "Sex Offender Registry",
"31": "Wanted Persons",
"32": "Booking Reports",
"33": "Court Cases",
"34": "Incarceration Records",
"35": "Poor Data Source"
},
"initializer_range": 0.02,
"label2id": {
"Accident Reports": 0,
"Annual & Monthly Reports": 19,
"Arrest Records": 1,
"Booking Reports": 32,
"Budgets & Finances": 20,
"Calls for Service": 2,
"Car GPS": 3,
"Citations": 4,
"Complaints & Misconduct": 15,
"Contact Info & Agency Meta": 21,
"Court Cases": 33,
"Crime Maps & Reports": 25,
"Crime Statistics": 26,
"Daily Activity Logs": 16,
"Dispatch Logs": 5,
"Dispatch Recordings": 6,
"Field Contacts": 7,
"Geographic": 22,
"Incarceration Records": 34,
"Incident Reports": 8,
"List of Data Sources": 23,
"Media Bulletins": 27,
"Misc Police Activity": 9,
"Officer Involved Shooting": 10,
"Personnel Records": 18,
"Policies & Contracts": 24,
"Poor Data Source": 35,
"Records Request Info": 28,
"Resources": 29,
"Sex Offender Registry": 30,
"Stops": 11,
"Surveys": 12,
"Training & Hiring Info": 17,
"Use of Force Reports": 13,
"Vehicle Pursuits": 14,
"Wanted Persons": 31
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.36.2",
"vocab_size": 30522
}