config.py 3.04 KB
Newer Older
1
2
from pathlib import Path

Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
3
PATHS_2019 = ['DigiRehab_BorgerID_TrainingDone.xlsx',
4
5
6
7
8
9
10
11
                    'DigiRehab_BorgerID_TrainingCancelled.xlsx',
                    'DigiRehab_BorgerID_StatusSet.xlsx',
                    'DigiRehab_BorgerID_ScreeningContent.xlsx',
                    'HjaelpemidlerUdtraek.csv',
                    'isoall.txt',
                    'DigiRehab_borgerIDALL.csv',
                    'clusters.csv']

Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
12
13
14
15
16
PATHS_2020 = ['borgere_hmi_Rasmus_BorgerId_Gender_BirthYear.xlsx',
              'DrPatientData_RasmusPlusBorgerIdMinusCPR_2020.xlsx',
              'Observationer_Rasmus_BorgerId_Gender_BirthYear.xlsx',
              'Træning_Rasmus_BorgerId_Gender_BirthYear.xlsx']

17
18
ROOT_DIR = Path(__file__).absolute().parent.parent
MODELS_DIR = Path.joinpath(ROOT_DIR, 'models')
19
REPORTS_DIR = Path.joinpath(ROOT_DIR, 'reports')
20
21
LOGS_DIR = Path.joinpath(ROOT_DIR, 'src/logs')
CONFIG_DIR = Path.joinpath(ROOT_DIR, 'src/cfg')
Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
22
23
24
TESTS_FILES_DIR = Path.joinpath(ROOT_DIR, 'tests/files')
RAW_DATA_DIR_2019 = Path.joinpath(ROOT_DIR, 'data/raw/2019')
RAW_DATA_DIR_2020 = Path.joinpath(ROOT_DIR, 'data/raw/2020')
25
26
27
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')
28
29
30
COMPLETE_DIR = Path.joinpath(ROOT_DIR, 'models/complete')
COMPLETE_TF_DIR = Path.joinpath(ROOT_DIR, 'models/complete/tensorflow')
COMPLETE_XGB_DIR = Path.joinpath(ROOT_DIR, 'models/complete/xgboost')
31
COMPLETE_CAT_DIR = Path.joinpath(ROOT_DIR, 'models/complete/catboost')
32
33
COMPLETE_RF_DIR = Path.joinpath(ROOT_DIR, 'models/complete/random_forest')
COMPLETE_EMB_DIR = Path.joinpath(ROOT_DIR, 'models/complete/embeddings')
34
COMPLETE_EMB_FULL_DIR = Path.joinpath(ROOT_DIR, 'models/complete/embeddings_full')
35
36
37
FALL_DIR = Path.joinpath(ROOT_DIR, 'models/fall')
FALL_TF_DIR = Path.joinpath(ROOT_DIR, 'models/fall/tensorflow')
FALL_XGB_DIR = Path.joinpath(ROOT_DIR, 'models/fall/xgboost')
38
FALL_CAT_DIR = Path.joinpath(ROOT_DIR, 'models/fall/catboost')
39
40
FALL_RF_DIR = Path.joinpath(ROOT_DIR, 'models/fall/random_forest')
FALL_EMB_DIR = Path.joinpath(ROOT_DIR, 'models/fall/embeddings')
Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
41
GENERAL_FEATURES = ['Gender', 'Age', 'Cluster']
42

43
44
THRESHOLD_WEEKS = 8
THRESHOLD_TRAINING = 10
45
COMPLETE_N_SCALE_COLS = 10
46
FALL_N_SCALE_COLS = 4
47

48
49
PATIENT_ID = 'PatientId'
CITIZEN_ID = 'CitizenId'
Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
50
GENDER = 'Gender'
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
BIRTH_YEAR = 'BirthYear'
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'
Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
71
72
73
74
DESCRIPTION = 'Description'
DATE = 'Date'
QUESTION = 'Question'
ANSWER = 'Answer'