risk-slim

risk-slim is a machine learning method to fit simple customized risk scores in python.

Background

Risk scores let users make quick risk predictions by adding and subtracting a few small numbers (see e.g., 500 + medical risk scores at mdcalc.com).

Here is a risk score for ICU risk prediction from our paper.

Video

Reference

If you use risk-slim in your research, we would appreciate a citation to the following paper (bibtex!

Learning Optimized Risk Scores
Berk Ustun and Cynthia Rudin
Journal of Machine Learning Research, 2019.

Installation

Run the following snippet in a Unix terminal to install risk-slim and complete a test run.

git clone https://github.com/ustunb/risk-slim
cd risk-slim
pip install -e .        # install in editable mode  
bash batch/job_template.sh  # batch run

Requirements

risk-slim requires Python 3.5+ and CPLEX 12.6+. For instructions on how to download and install, click here.

Contributing

I'm planning to pick up development again in Fall 2020. I can definitely use a hand! If you are interested in contributing, please reach out!

Here's the current development roadmap:

• sci-kit learn interface
• support for open source solver in python-mip
• basic reporting tools (roc curves, calibration plots, model reports)
• documentation
• pip