config.py 4.5 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from pathlib import Path

FILE_PATHS = ['DigiRehab_BorgerID_TrainingDone.xlsx',
                    'DigiRehab_BorgerID_TrainingCancelled.xlsx',
                    'DigiRehab_BorgerID_StatusSet.xlsx',
                    'DigiRehab_BorgerID_ScreeningContent.xlsx',
                    'HjaelpemidlerUdtraek.csv',
                    'isoall.txt',
                    'DigiRehab_borgerIDALL.csv',
                    'clusters.csv']

ROOT_DIR = Path(__file__).absolute().parent.parent
MODELS_DIR = Path.joinpath(ROOT_DIR, 'models')
LOGS_DIR = Path.joinpath(ROOT_DIR, 'src/logs')
CONFIG_DIR = Path.joinpath(ROOT_DIR, 'src/cfg')
TESTS_FILES_DIR = Path.joinpath(ROOT_DIR, 'src/tests/files')
RAW_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/raw')
PROCESSED_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/processed')
INTERIM_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/interim')
EXTERNAL_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/external')

22
GENERAL_FEATURES = ['Sex', 'Age', 'Cluster']
23

24
DEVICE_FEATURES = ['HasRollator',
25
                 'HasShowerStool',
26
27
28
29
30
31
                 'HasRaisedToiletSeat',
                 'HasEmergencySystem',
                 'HasSeatCushion',
                 'HasWheelchair',
                 'HasBedWithEngine',
                 'ATS']
32

33
34
COMPLETES_FEATURES = ['NumberWeeksSum', 'NumberTrainingSum']

35
TRAINING_FEATURES = ['NeedsStartBaseline',
36
37
38
39
40
41
42
                 'MeanEvaluationMean',
                 'StdEvaluationMean',
                 'NumberTrainingWeekMean',
                 'MeanTimeBetweenTrainingMean',
                 'NumberCancelsSum',
                 'MeanTimeBetweenCancelsMean',
                 'MeanNumberCancelsWeekMean',
43
44
45
46
47
                 'NeedsDifferenceMean',
                 'PhysicsDifferenceMean',
                 'NumberExercisesMean']

ATS_FEATURES = ['NumberAtsMean',
48
49
50
51
52
53
                 'HadRollator',
                 'HadShowerStool',
                 'HadRaisedToiletSeat',
                 'HadEmergencySystem',
                 'HadSeatCushion',
                 'HadWheelchair',
54
                 'HadBedWithEngine']
55

56
TRAINING_FEATURES_SCR = ['NumberWeeks',
57
58
59
60
                 'MeanEvaluation',
                 'StdEvaluation',
                 'MinEvaluation',
                 'MaxEvaluation',
61
62
63
64
65
66
67
68
69
70
71
72
73
                 'NumberTraining',
                 'NumberTrainingPerWeek',
                 'NumberTrainingPerWeekMax',
                 'NumberTrainingPerWeekMin',
                 'NumberWeeksWithTraining',
                 'TimeBetweenTrainingAvg',
                 'NumberWeeksNoTraining',
                 'NumberCancels',
                 'TimeBetweenCancelsAvg',
                 'NumberCancelsPerWeekAvg',
                 'NumberCancelsPerWeekMin',
                 'NumberCancelsPerWeekMax',
                 'NumberExercises']
74

75
OBJECT_COLS = ['HasRollator',
76
               'HasRaisedToiletSeat',
77
               'HasShowerStool']
78

79
DATE_COLS = ['StartDate', 'EndDate', 'LastStatusDate'] # TODO: Decide about LastStatusDate
80
LIST_COLS = ['ATS']
81

82
TWO_SCREENINGS_SUBSET = ['StartYear',
83
                        'NumberAts',
84
85
86
87
88
89
90
91
92
93
94
95
                        'Sex',
                        'RehabIndicator',
                        'NeedsStart',
                        'PhysicsStart',
                        'NeedsDifference',
                        'MeanEvaluation',
                        'StdEvaluation',
                        'MinEvaluation',
                        'MaxEvaluation',
                        'nTrainingPrWeek',
                        'nTrainingPrWeekMax',
                        'nTrainingPrWeekMin',
96
                        'TimeBetweenTrainingAvg',
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
                        'nCancellationsPrWeekAVG',
                        'nCancellationsPrWeekMax',
                        'nCancellationsPrWeekMin',
                        'NeedsEnd',
                        'PhysicsEnd',
                        'PhysicsDifference']

PATIENT_ID = 'PatientId'
CITIZEN_ID = 'CitizenId'
BIRTH_YEAR = 'BirthYear'
SEX = 'Sex'
RATING_DATE = 'RatingDate'
RATING_SCORE = 'RatingScore'
CHANGE_DATE = 'ChangeDate'
STATUS = 'Status'
SCREENING_DATE = 'ScreeningDate'
NEED_FOR_HELP_SCORE = 'NeedForHelpScore'
NEED_FOR_HELP_REASON = 'NeedForHelpReason'
PHYSICAL_STRENGTH_SCORE = 'PhysicalStrengthScore'
PHYSICAL_STRENGTH_REASON = 'PhysicalStrengthReason'
EXERCISE_CONTENT = 'ExerciseContent'
DEV_HMI_NUMBER = 'DevHMINumber'
DEV_HMI_NAME   = 'DevHMIName'
DEV_ISO_CLASS  = 'DevISOClass'
DEV_SERIAL    = 'DevSerial'
LAW_PARAGRAPH = 'LawParagraph'
LEND_DATE     = 'LendDate'
RETURN_DATE   = 'ReturnDate'
PRICE        = 'Price'
GROUP_SIZE   = 'GroupSize'
DESCRIPTION = 'Description'