/PEARL

deep prototype learning for EHR data

Primary LanguagePython

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)