futianfan / PEARL

deep prototype learning for EHR data
https://futianfan.github.io/BCB19_PEARL.pdf
3 stars 0 forks source link
deep-learning healthcare interpretable-machine-learning phenotyping prototype-lear rule-based-modeling

PEARL: Prototype Learning via Rule Learning

🚀 Installation

conda create -n pearl python=3.7 
conda activate pearl 
pip install torch 
pip install scikit-learn 
pip install matplotlib 

📊 Data

data/: data folder

src/stream.py: process data

⚙️ Configuration

src/config.py: configuration

🤖 Model

src/decision_tree.py: decision-tree for interpretable rule

src/model_torch.py: NN model (Pytorch)

train

python src/train_torch.py

📞 Contact

Tianfan Fu (futianfan@gmail.com)