{ "_name_or_path": "AutoTrain", "_num_labels": 20, "architectures": [ "BertForSequenceClassification" ], "attention_probs_dropout_prob": 0.1, "classifier_dropout": null, "directionality": "bidi", "gradient_checkpointing": false, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 1024, "id2label": { "0": "Certain infectious or parasitic diseases", "1": "Developmental anaomalies", "2": "Diseases of the blood or blood forming organs", "3": "Diseases of the genitourinary system", "4": "Mental behavioural or neurodevelopmental disorders", "5": "Neoplasms", "6": "certain conditions originating in the perinatal period", "7": "conditions related to sexual health", "8": "diseases of the circulatroy system", "9": "diseases of the digestive system", "10": "diseases of the ear or mastoid process", "11": "diseases of the immune system", "12": "diseases of the musculoskeletal system or connective tissue", "13": "diseases of the nervous system", "14": "diseases of the respiratory system", "15": "diseases of the skin", "16": "diseases of the visual system", "17": "endocrine nutritional or metabolic diseases", "18": "pregnanacy childbirth or the puerperium", "19": "sleep-wake disorders" }, "initializer_range": 0.02, "intermediate_size": 4096, "label2id": { "Certain infectious or parasitic diseases": 0, "Developmental anaomalies": 1, "Diseases of the blood or blood forming organs": 2, "Diseases of the genitourinary system": 3, "Mental behavioural or neurodevelopmental disorders": 4, "Neoplasms": 5, "certain conditions originating in the perinatal period": 6, "conditions related to sexual health": 7, "diseases of the circulatroy system": 8, "diseases of the digestive system": 9, "diseases of the ear or mastoid process": 10, "diseases of the immune system": 11, "diseases of the musculoskeletal system or connective tissue": 12, "diseases of the nervous system": 13, "diseases of the respiratory system": 14, "diseases of the skin": 15, "diseases of the visual system": 16, "endocrine nutritional or metabolic diseases": 17, "pregnanacy childbirth or the puerperium": 18, "sleep-wake disorders": 19 }, "layer_norm_eps": 1e-12, "max_length": 128, "max_position_embeddings": 512, "model_type": "bert", "num_attention_heads": 16, "num_hidden_layers": 24, "pad_token_id": 0, "padding": "max_length", "pooler_fc_size": 768, "pooler_num_attention_heads": 12, "pooler_num_fc_layers": 3, "pooler_size_per_head": 128, "pooler_type": "first_token_transform", "position_embedding_type": "absolute", "problem_type": "single_label_classification", "torch_dtype": "float32", "transformers_version": "4.15.0", "type_vocab_size": 2, "use_cache": true, "vocab_size": 28996 }