book-genre-classification / head_config.json
davanstrien's picture
davanstrien HF staff
add model
3953f0d
{
"config": {
"activation_function": "tanh",
"bias": true,
"head_type": "classification",
"label2id": {
"Arts & Photography": 12,
"Biographies & Memoirs": 22,
"Business & Money": 20,
"Calendars": 0,
"Children's Books": 30,
"Christian Books & Bibles": 26,
"Comics & Graphic Novels": 1,
"Computers & Technology": 16,
"Cookbooks, Food & Wine": 10,
"Crafts, Hobbies & Home": 11,
"Education & Teaching": 13,
"Engineering & Transportation": 9,
"Gay & Lesbian": 8,
"Health, Fitness & Dieting": 19,
"History": 23,
"Humor & Entertainment": 6,
"Law": 21,
"Literature & Fiction": 7,
"Medical Books": 17,
"Mystery, Thriller & Suspense": 3,
"Parenting & Relationships": 14,
"Politics & Social Sciences": 24,
"Reference": 25,
"Religion & Spirituality": 27,
"Romance": 5,
"Science & Math": 18,
"Science Fiction & Fantasy": 4,
"Self-Help": 15,
"Sports & Outdoors": 28,
"Teen & Young Adult": 29,
"Test Preparation": 2,
"Travel": 31
},
"layers": 2,
"num_labels": 32,
"use_pooler": false
},
"hidden_size": 768,
"model_class": "BertModelWithHeads",
"model_name": "bert-base-cased",
"model_type": "bert",
"name": "contemporary-genre"
}