Add config.json

config.json

@@ -1,11 +1,10 @@
 {
-  "model_type": "autism_detector",
-  "architecture": "AutismDetectorNet",
-  "input_size": 202,
+  "model_type": "autism_detector_simplified",
+  "architecture": "SimplifiedASDDetector",
+  "input_size": 8,
   "hidden_sizes": [
-    128,
-    64,
-    32
+    32,
+    16
   ],
   "dropout_rate": 0.3,
   "num_classes": 2,
@@ -13,296 +12,75 @@
     "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"
-  ],
+  "num_features": 8,
+  "features": {
+    "original_names": [
+      "Developmental milestones- global delay (G), motor delay (M), cognitive delay (C)",
+      "IQ/DQ",
+      "ICD",
+      "Language disorder Y= present, N=absent",
+      "Language development: delay, normal=N, absent=A",
+      "Dysmorphysm y=present, no=absent",
+      "Behaviour disorder- agressivity, agitation, irascibility",
+      "Neurological Examination; N=normal, text = abnormal; free cell = examination not performed ???"
+    ],
+    "simple_names": [
+      "developmental_milestones",
+      "iq_dq",
+      "intellectual_disability",
+      "language_disorder",
+      "language_development",
+      "dysmorphism",
+      "behaviour_disorder",
+      "neurological_exam"
+    ],
+    "value_mappings": {
+      "developmental_milestones": {
+        "N": "Normal",
+        "G": "Global delay",
+        "M": "Motor delay",
+        "C": "Cognitive delay"
+      },
+      "iq_dq": "numeric (0-150)",
+      "intellectual_disability": {
+        "N": "None",
+        "F70.0": "Mild",
+        "F71": "Moderate",
+        "F72": "Severe"
+      },
+      "language_disorder": {
+        "N": "No",
+        "Y": "Yes"
+      },
+      "language_development": {
+        "N": "Normal",
+        "delay": "Delayed",
+        "A": "Absent"
+      },
+      "dysmorphism": {
+        "NO": "No",
+        "Y": "Yes"
+      },
+      "behaviour_disorder": {
+        "N": "No",
+        "Y": "Yes"
+      },
+      "neurological_exam": {
+        "N": "Normal",
+        "other": "Abnormal (describe)"
+      }
+    }
+  },
   "metrics": {
-    "test_recall": 0.9682539682539683,
+    "test_recall": 0.9365079365079365,
     "test_precision": 1.0,
-    "test_f1": 0.9838709677419355,
-    "test_accuracy": 0.9759036144578314,
-    "test_auc": 0.9944444444444445
+    "test_f1": 0.9672131147540983,
+    "test_accuracy": 0.9518072289156626,
+    "test_auc": 0.9904761904761905
   },
   "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
+    "feature_importance_coverage": "84.2%"
   }
 }