Add config.json

config.json

@@ -0,0 +1,308 @@
+{
+  "model_type": "autism_detector",
+  "architecture": "AutismDetectorNet",
+  "input_size": 202,
+  "hidden_sizes": [
+    128,
+    64,
+    32
+  ],
+  "dropout_rate": 0.3,
+  "num_classes": 2,
+  "class_labels": [
+    "Healthy",
+    "ASD"
+  ],
+  "feature_columns": [
+    "Gender",
+    "Age",
+    "Urban/Rural",
+    "Pregnancy natural /IVF",
+    "Single/twins",
+    "Pregnancy evolution, N=normal, AN=abnormal",
+    "Birth weeks",
+    "Type of birth",
+    "Birth Weight g",
+    "Length at birth  ",
+    "Head circumference at birth  ",
+    "APGAR score",
+    "Postnatal adaptation N=normal, AN=abnormal",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)",
+    "Other Chronic diseases",
+    "Infections",
+    "allergies",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent",
+    "Mother age (years)",
+    "Father age (years)",
+    "IQ/DQ",
+    "ICD",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???",
+    "Weight kg",
+    "Height  ",
+    "head circumf  ",
+    "Dysmorphysm y=present, no=absent",
+    "malformations Y= present, N=absent",
+    "Behaviour disorder- agressivity, agitation, irascibility",
+    "Language development: delay, normal=N, absent=A",
+    "Language disorder Y= present, N=absent",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges",
+    "MRI structural anomalies of the brain, N=absent, AN=present"
+  ],
+  "numeric_columns": [
+    "Age",
+    "Birth weeks",
+    "Birth Weight g",
+    "Length at birth  ",
+    "Head circumference at birth  ",
+    "APGAR score",
+    "Infections",
+    "allergies",
+    "Mother age (years)",
+    "Father age (years)",
+    "Weight kg",
+    "Height  ",
+    "head circumf  "
+  ],
+  "categorical_columns": [
+    "Gender",
+    "Urban/Rural",
+    "Pregnancy natural /IVF",
+    "Single/twins",
+    "Pregnancy evolution, N=normal, AN=abnormal",
+    "Type of birth",
+    "Postnatal adaptation N=normal, AN=abnormal",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)",
+    "Other Chronic diseases",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent",
+    "IQ/DQ",
+    "ICD",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???",
+    "Dysmorphysm y=present, no=absent",
+    "malformations Y= present, N=absent",
+    "Behaviour disorder- agressivity, agitation, irascibility",
+    "Language development: delay, normal=N, absent=A",
+    "Language disorder Y= present, N=absent",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges",
+    "MRI structural anomalies of the brain, N=absent, AN=present"
+  ],
+  "transformed_feature_names": [
+    "Age",
+    "Birth weeks",
+    "Birth Weight g",
+    "Length at birth  ",
+    "Head circumference at birth  ",
+    "APGAR score",
+    "Infections",
+    "allergies",
+    "Mother age (years)",
+    "Father age (years)",
+    "Weight kg",
+    "Height  ",
+    "head circumf  ",
+    "Gender_F",
+    "Gender_F ",
+    "Gender_M",
+    "Gender_M ",
+    "Urban/Rural_R",
+    "Urban/Rural_U",
+    "Urban/Rural_missing",
+    "Urban/Rural_u",
+    "Pregnancy natural /IVF_IVF",
+    "Pregnancy natural /IVF_N",
+    "Single/twins_S",
+    "Single/twins_TWINS",
+    "Pregnancy evolution, N=normal, AN=abnormal_AN",
+    "Pregnancy evolution, N=normal, AN=abnormal_N",
+    "Pregnancy evolution, N=normal, AN=abnormal_missing",
+    "Type of birth_C",
+    "Type of birth_N",
+    "Type of birth_c",
+    "Type of birth_missing",
+    "Postnatal adaptation N=normal, AN=abnormal_AN",
+    "Postnatal adaptation N=normal, AN=abnormal_N",
+    "Postnatal adaptation N=normal, AN=abnormal_missing",
+    "Postnatal adaptation N=normal, AN=abnormal_n",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_C",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_G",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_M",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_N",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_c",
+    "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)_missing",
+    "Other Chronic diseases_N",
+    "Other Chronic diseases_Y",
+    "Other Chronic diseases_missing",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent_NE",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent_NO",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent_P",
+    "Family history psychiatric (P) or Neuro disease (Ne), No=absent_missing",
+    "IQ/DQ_20",
+    "IQ/DQ_22",
+    "IQ/DQ_25",
+    "IQ/DQ_30",
+    "IQ/DQ_32",
+    "IQ/DQ_33",
+    "IQ/DQ_34",
+    "IQ/DQ_35",
+    "IQ/DQ_36",
+    "IQ/DQ_38",
+    "IQ/DQ_39",
+    "IQ/DQ_40",
+    "IQ/DQ_41",
+    "IQ/DQ_42",
+    "IQ/DQ_43",
+    "IQ/DQ_44",
+    "IQ/DQ_45",
+    "IQ/DQ_46",
+    "IQ/DQ_47",
+    "IQ/DQ_48",
+    "IQ/DQ_49",
+    "IQ/DQ_50",
+    "IQ/DQ_51",
+    "IQ/DQ_52",
+    "IQ/DQ_53",
+    "IQ/DQ_54",
+    "IQ/DQ_55",
+    "IQ/DQ_56",
+    "IQ/DQ_57",
+    "IQ/DQ_59",
+    "IQ/DQ_60",
+    "IQ/DQ_61",
+    "IQ/DQ_62",
+    "IQ/DQ_65",
+    "IQ/DQ_66",
+    "IQ/DQ_67",
+    "IQ/DQ_68",
+    "IQ/DQ_69",
+    "IQ/DQ_70",
+    "IQ/DQ_72",
+    "IQ/DQ_74",
+    "IQ/DQ_75",
+    "IQ/DQ_76",
+    "IQ/DQ_77",
+    "IQ/DQ_78",
+    "IQ/DQ_79",
+    "IQ/DQ_80",
+    "IQ/DQ_82",
+    "IQ/DQ_83",
+    "IQ/DQ_84",
+    "IQ/DQ_85",
+    "IQ/DQ_86",
+    "IQ/DQ_87",
+    "IQ/DQ_88",
+    "IQ/DQ_89",
+    "IQ/DQ_90",
+    "IQ/DQ_91",
+    "IQ/DQ_92",
+    "IQ/DQ_94",
+    "IQ/DQ_95",
+    "IQ/DQ_96",
+    "IQ/DQ_97",
+    "IQ/DQ_98",
+    "IQ/DQ_100",
+    "IQ/DQ_101",
+    "IQ/DQ_102",
+    "IQ/DQ_103",
+    "IQ/DQ_104",
+    "IQ/DQ_105",
+    "IQ/DQ_106",
+    "IQ/DQ_107",
+    "IQ/DQ_108",
+    "IQ/DQ_110",
+    "IQ/DQ_112",
+    "IQ/DQ_113",
+    "IQ/DQ_114",
+    "IQ/DQ_115",
+    "IQ/DQ_120",
+    "IQ/DQ_122",
+    "IQ/DQ_124",
+    "IQ/DQ_126",
+    "IQ/DQ_127",
+    "IQ/DQ_128",
+    "IQ/DQ_130",
+    "IQ/DQ_132",
+    "IQ/DQ_135",
+    "IQ/DQ_136",
+    "IQ/DQ_138",
+    "IQ/DQ_150",
+    "ICD_F70.0",
+    "ICD_F71",
+    "ICD_F72",
+    "ICD_N",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_Convergent squint both eyes",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_Impaired fine motor skills",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_Inability to perform fine movements",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_N",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_N\n",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_cerebral palsy",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_cerebral palsy, tetraparesis",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_convergent strabismus, hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_convergent strabysmus of the OA, right eye>left eye; inability of movement",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine and gross motor skills",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine and gross motor skills ",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine and gross motor skills; hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine motor skills",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine motor skills; hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine motor skills; hypotonia; ",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_difficulty with fine motor skills; mild muscular hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_extrapyramidal movements; generalized hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hyotonia\n",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hypersensitivity to sounds",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hypotonia, fine movements inability, mild tremor when grabbing object",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hypotonia; difficulty with fine and gross motor skills ",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_hypotonia; inability to walk",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_impaired fine motor skills",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_inability for fine and gross motor skills; hypotonia ",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_inability for fine movements.",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_inability of the fine and gross motor skills; hypotonia",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_inability to perform fine movements",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_inability to perform fine movements; hypotonia ",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_joint hyperlaxity",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_ligamentar hiperlaxity",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_mild motor inability",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_mild muscle hypotonia; no sphincter control",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_mild muscular hypotonia; inability to perform fine movements",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_missing",
+    "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???_muscular hypotnia",
+    "Dysmorphysm y=present, no=absent_NO",
+    "Dysmorphysm y=present, no=absent_Y",
+    "malformations Y= present, N=absent_N",
+    "malformations Y= present, N=absent_Y",
+    "malformations Y= present, N=absent_missing",
+    "Behaviour disorder- agressivity, agitation, irascibility_N",
+    "Behaviour disorder- agressivity, agitation, irascibility_Y",
+    "Behaviour disorder- agressivity, agitation, irascibility_Y\n",
+    "Language development: delay, normal=N, absent=A_A",
+    "Language development: delay, normal=N, absent=A_N",
+    "Language development: delay, normal=N, absent=A_Y",
+    "Language development: delay, normal=N, absent=A_delay",
+    "Language development: delay, normal=N, absent=A_delay\n",
+    "Language disorder Y= present, N=absent_N",
+    "Language disorder Y= present, N=absent_Y",
+    "Language disorder Y= present, N=absent_missing",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges_F",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges_G",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges_N",
+    "EEG, N=normal, F=focal discharges, G=bilateral discharges_missing",
+    "MRI structural anomalies of the brain, N=absent, AN=present_AN",
+    "MRI structural anomalies of the brain, N=absent, AN=present_AN\n",
+    "MRI structural anomalies of the brain, N=absent, AN=present_N",
+    "MRI structural anomalies of the brain, N=absent, AN=present_missing"
+  ],
+  "metrics": {
+    "test_recall": 0.9682539682539683,
+    "test_precision": 1.0,
+    "test_f1": 0.9838709677419355,
+    "test_accuracy": 0.9759036144578314,
+    "test_auc": 0.9944444444444445
+  },
+  "training_info": {
+    "framework": "pytorch",
+    "torch_version": "2.9.1+cu128",
+    "device": "cuda",
+    "batch_size": 32,
+    "learning_rate": 0.001,
+    "max_epochs": 100,
+    "early_stopping_patience": 15
+  }
+}