ConMedRL 1.0.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- conmedrl-1.0.0/ConMedRL/__init__.py +51 -0
- conmedrl-1.0.0/ConMedRL/conmedrl.py +1752 -0
- conmedrl-1.0.0/ConMedRL/conmedrl_continuous.py +2123 -0
- conmedrl-1.0.0/ConMedRL/data_loader.py +154 -0
- conmedrl-1.0.0/ConMedRL.egg-info/PKG-INFO +561 -0
- conmedrl-1.0.0/ConMedRL.egg-info/SOURCES.txt +25 -0
- conmedrl-1.0.0/ConMedRL.egg-info/dependency_links.txt +1 -0
- conmedrl-1.0.0/ConMedRL.egg-info/not-zip-safe +1 -0
- conmedrl-1.0.0/ConMedRL.egg-info/requires.txt +28 -0
- conmedrl-1.0.0/ConMedRL.egg-info/top_level.txt +2 -0
- conmedrl-1.0.0/Data/SICdb_discharge/__init__.py +7 -0
- conmedrl-1.0.0/Data/SICdb_discharge/data_preprocess.py +1534 -0
- conmedrl-1.0.0/Data/SICdb_extubation/__init__.py +7 -0
- conmedrl-1.0.0/Data/SICdb_extubation/data_preprocess.py +1612 -0
- conmedrl-1.0.0/Data/__init__.py +24 -0
- conmedrl-1.0.0/Data/mimic_iv_icu_discharge/__init__.py +7 -0
- conmedrl-1.0.0/Data/mimic_iv_icu_discharge/data_preprocess.py +1478 -0
- conmedrl-1.0.0/Data/mimic_iv_icu_extubation/__init__.py +7 -0
- conmedrl-1.0.0/Data/mimic_iv_icu_extubation/data_preprocess.py +1614 -0
- conmedrl-1.0.0/LICENSE +21 -0
- conmedrl-1.0.0/MANIFEST.in +33 -0
- conmedrl-1.0.0/PKG-INFO +561 -0
- conmedrl-1.0.0/README.md +483 -0
- conmedrl-1.0.0/pyproject.toml +102 -0
- conmedrl-1.0.0/requirements.txt +17 -0
- conmedrl-1.0.0/setup.cfg +4 -0
- conmedrl-1.0.0/setup.py +101 -0
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
"""
|
|
2
|
+
ConCare-RL: An Offline Constrained Reinforcement Learning Toolkit for Critical Care Decision Making
|
|
3
|
+
|
|
4
|
+
This package provides the core OCRL framework for critical care decision support,
|
|
5
|
+
including policy evaluation, policy optimization, and data handling utilities.
|
|
6
|
+
|
|
7
|
+
Authors: Maotong Sun (maotong.sun@tum.de), Jingui Xie (jingui.xie@tum.de)
|
|
8
|
+
"""
|
|
9
|
+
|
|
10
|
+
__version__ = "1.0.0"
|
|
11
|
+
__author__ = "Maotong Sun, Jingui Xie"
|
|
12
|
+
__email__ = "maotong.sun@tum.de, jingui.xie@tum.de"
|
|
13
|
+
|
|
14
|
+
# Core OCRL components
|
|
15
|
+
from .conmedrl import (
|
|
16
|
+
FCN_fqe,
|
|
17
|
+
FCN_fqi,
|
|
18
|
+
ReplayBuffer,
|
|
19
|
+
FQE,
|
|
20
|
+
FQI,
|
|
21
|
+
RLConfig_custom,
|
|
22
|
+
RLConfigurator,
|
|
23
|
+
RLTraining
|
|
24
|
+
)
|
|
25
|
+
|
|
26
|
+
# Data loading utilities
|
|
27
|
+
from .data_loader import (
|
|
28
|
+
TrainDataLoader,
|
|
29
|
+
ValTestDataLoader
|
|
30
|
+
)
|
|
31
|
+
|
|
32
|
+
__all__ = [
|
|
33
|
+
# Core OCRL classes
|
|
34
|
+
'FCN_fqe',
|
|
35
|
+
'FCN_fqi',
|
|
36
|
+
'ReplayBuffer',
|
|
37
|
+
'FQE',
|
|
38
|
+
'FQI',
|
|
39
|
+
'RLConfig_custom',
|
|
40
|
+
'RLConfigurator',
|
|
41
|
+
'RLTraining',
|
|
42
|
+
|
|
43
|
+
# Data loading classes
|
|
44
|
+
'TrainDataLoader',
|
|
45
|
+
'ValTestDataLoader',
|
|
46
|
+
|
|
47
|
+
# Package metadata
|
|
48
|
+
'__version__',
|
|
49
|
+
'__author__',
|
|
50
|
+
'__email__'
|
|
51
|
+
]
|