This repository contains Python code implementing the Rational Multi-Layer Perceptrons (RMLP) model, along with experimentation on the MIMIC-III dataset. This implementation accompanies the paper:
Suttaket, T., & Kok, S. (2024). Interpretable Predictive Models for Healthcare via Rational Multi-Layer Perceptrons. ACM Transactions on Management Information Systems.
- Download the MIMIC-III dataset by following the instructions provided at MIMIC-III Getting Started.
- Create two tasks which are the in-hospital mortality and decompensation prediction tasks by following the instructions from the MIMIC-III Benchmark.
To train the RMLP model for the in hospital mortality task, run the following command:
```sh
python -u main_ihm.py --log_likelihood_fn=loglikelihood_filename --deep_supervision=1 --pattern_specs="3-7_6-4_9-7_12-7_15-2_18-7" --batch_size=128 --data=data_folder --dropout=0.0 --clip=0 --mlp_hidden_dim=0 --num_mlp_layers=1 --mlp_pattern_NN="30" --target_repl_coef=0.5 --epochs=200 --file_name=output_filename
```
To train the RMLP model for the decompensation task, run the following command:
```sh
python -u main_decomp.py --log_likelihood_fn=loglikelihood_filename --pattern_specs='3-10_4-10_5-10' --mlp_hidden_dim=0 --num_mlp_layers=1 --dropout=0 --batch_size=25 --file_name=output_filename --epochs=0 --mlp_pattern_NN="30-10"
```