final_bert_topic_model / config.json
metechmohit's picture
Upload all model and tokenizer files
fef5f20 verified
{
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"dtype": "float32",
"hidden_dim": 3072,
"id2label": {
"0": "History - Ancient India",
"1": "History - Medieval India",
"2": "History - Modern India",
"3": "History - World",
"4": "Polity & Governance - India",
"5": "Polity & Governance - Comparative",
"6": "Geography - India",
"7": "Geography - World",
"8": "Economy - India",
"9": "Economy - Global",
"10": "International Relations - Bilateral",
"11": "International Relations - Global Groupings",
"12": "Science & Technology - Core Concepts",
"13": "Science & Technology - Space & Defense",
"14": "General Query - Meta",
"15": "Science & Technology - IT & Communication",
"16": "Environment & Ecology - Core Concepts",
"17": "Environment & Ecology - Indian Context",
"18": "Current Affairs - National",
"19": "Current Affairs - International"
},
"initializer_range": 0.02,
"label2id": {
"Current Affairs - International": 19,
"Current Affairs - National": 18,
"Economy - Global": 9,
"Economy - India": 8,
"Environment & Ecology - Core Concepts": 16,
"Environment & Ecology - Indian Context": 17,
"General Query - Meta": 14,
"Geography - India": 6,
"Geography - World": 7,
"History - Ancient India": 0,
"History - Medieval India": 1,
"History - Modern India": 2,
"History - World": 3,
"International Relations - Bilateral": 10,
"International Relations - Global Groupings": 11,
"Polity & Governance - Comparative": 5,
"Polity & Governance - India": 4,
"Science & Technology - Core Concepts": 12,
"Science & Technology - IT & Communication": 15,
"Science & Technology - Space & Defense": 13
},
"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,
"transformers_version": "4.57.1",
"vocab_size": 30522
}