functional-effects-model/src/constants.py at main · big-ucl/functional-effects-model · 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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
PATH_TO_DATA = {
    "easySHARE": "../data/easySHARE/easySHARE_preprocessed.csv",
    "SwissMetro": "../data/SwissMetro/train.pkl",
    "LPMC": "../data/LPMC/",
}
PATH_TO_DATA_TRAIN = {
    "easySHARE": "../data/easySHARE/easySHARE_preprocessed_train.csv",
    "SwissMetro": "../data/SwissMetro/train.pkl",
    "LPMC": "../data/LPMC/LPMC_train.csv",
}
PATH_TO_DATA_VAL = {
    "easySHARE": "../data/easySHARE/easySHARE_preprocessed_val.csv",
    "SwissMetro": "../data/SwissMetro/dev.pkl",
    "LPMC": "",
}
PATH_TO_DATA_TEST = {
    "easySHARE": "../data/easySHARE/easySHARE_preprocessed_test.csv",
    "SwissMetro": "../data/SwissMetro/test.pkl",
    "LPMC": "../data/LPMC/LPMC_test.csv",
}
PATH_TO_FOLDS = {
    "easySHARE": "../data/easySHARE/easySHARE_preprocessed_folds.pickle",
    "SwissMetro": "",
    "LPMC": "../data/LPMC/strat_group_k_fold_london.pickle",
}

alt_spec_features = {
    "easySHARE": {
        0: [
            "chronic_mod",
            "nb_doctor_visits",
            "maxgrip",
            "daily_activities_index",
            "instrumental_activities_index",
            "mobilityind",
            "lgmuscle",
            "grossmotor",
            "finemotor",
            "recall_1",
            "recall_2",
            "bmi",
            # "sphus_excellent", normalised to 0
            "sphus_fair",
            "sphus_good",
            "sphus_poor",
            "sphus_very_good",
            "hospitalised_last_year_yes",
            # "hospitalised_last_year_no", normalised to 0
            "nursing_home_last_year_yes_permanently",
            "nursing_home_last_year_yes_temporarily",
            # "nursing_home_last_year_no", normalised to 0
        ]
    },
    "SwissMetro": {
        0: [
            "TRAIN_TT",
            "TRAIN_HE",
            "TRAIN_CO",
        ],
        1: [
            "SM_TT",
            "SM_HE",
            "SM_CO",
            "SM_SEATS",
        ],
        2: [
            "CAR_TT",
            "CAR_CO",
        ],
    },
    "LPMC": {
        0: [
            "dur_walking",
            "distance",
            "day_of_week",
            "start_time_linear",
        ],
        1: [
            "dur_cycling",
            "distance",
            "day_of_week",
            "start_time_linear",
        ],
        2: [
            "dur_pt_access",
            "dur_pt_rail",
            "dur_pt_bus",
            "dur_pt_int_waiting",
            "dur_pt_int_walking",
            "pt_n_interchanges",
            "cost_transit",
            "distance",
            "day_of_week",
            "start_time_linear",
        ],
        3: [
            "dur_driving",
            "cost_driving_fuel",
            "congestion_charge",
            "distance",
            "driving_traffic_percent",
            "day_of_week",
            "start_time_linear",
        ],
    },
}