config.py 2.94 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
19
20
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')
Christian Marius Lillelund's avatar
Christian Marius Lillelund committed
21
22
23
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')
24
25
26
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')
27
28
29
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')
30
COMPLETE_CAT_DIR = Path.joinpath(ROOT_DIR, 'models/complete/catboost')
31
32
COMPLETE_RF_DIR = Path.joinpath(ROOT_DIR, 'models/complete/random_forest')
COMPLETE_EMB_DIR = Path.joinpath(ROOT_DIR, 'models/complete/embeddings')
33
COMPLETE_EMB_FULL_DIR = Path.joinpath(ROOT_DIR, 'models/complete/embeddings_full')
34
35
36
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')
37
FALL_CAT_DIR = Path.joinpath(ROOT_DIR, 'models/fall/catboost')
38
39
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
40
GENERAL_FEATURES = ['Gender', 'Age', 'Cluster']
41

42
43
44
THRESHOLD_WEEKS = 8
THRESHOLD_TRAINING = 10

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