TabDiff/data/Info/diabetes_dcr.json at main · MinkaiXu/TabDiff · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
{
    "name": "diabetes_dcr",
    "task_type": "binclass",
    "header": "infer",
    "column_names": [
        "num_lab_procedures",
        "num_medications",
        "number_outpatient",
        "number_emergency",
        "number_inpatient",
        "age",
        "time_in_hospital",
        "num_procedures",
        "number_diagnoses",
        "race",
        "gender",
        "admission_type_id",
        "discharge_disposition_id",
        "admission_source_id",
        "diag_1",
        "diag_2",
        "diag_3",
        "max_glu_serum",
        "A1Cresult",
        "metformin",
        "repaglinide",
        "nateglinide",
        "chlorpropamide",
        "glimepiride",
        "glipizide",
        "glyburide",
        "tolbutamide",
        "pioglitazone",
        "rosiglitazone",
        "acarbose",
        "miglitol",
        "tolazamide",
        "insulin",
        "glyburide-metformin",
        "change",
        "diabetesMed",
        "readmitted"
    ],
    "file_type": "csv",
    "raw_data_path": "data/diabetes_dcr/diabetic_data.csv",
    "data_path": "data/diabetes_dcr/diabetic_train.csv",
    "val_path": "data/diabetes_dcr/diabetic_val.csv",
    "test_path": "data/diabetes_dcr/diabetic_test.csv",
    "num_col_idx": [
        0,
        1,
        2,
        3,
        4,
        5,
        6,
        7,
        8
    ],
    "cat_col_idx": [
        9,
        10,
        11,
        12,
        13,
        14,
        15,
        16,
        17,
        18,
        19,
        20,
        21,
        22,
        23,
        24,
        25,
        26,
        27,
        28,
        29,
        30,
        31,
        32,
        33,
        34,
        35
    ],
    "target_col_idx": [
        36
    ]
}