Code Health Appveyor build status Build Status

PyPI version DOI GitHub license

The aim of healthcareai is to streamline machine learning in healthcare. The package has two main goals:



Why Anaconda?

We recommend using the Anaconda python distribution when working on Windows. There are a number of reasons:


You may need to install the following dependencies:

Once you have the dependencies satisfied run pip install healthcareai or sudo pip install healthcareai


Linux and macOS (via docker)

Verify Installation

To verify that healthcareai installed correctly, open a terminal and run python. This opens an interactive python console (also known as a REPL). Then enter this command: from healthcareai import SupervisedModelTrainer and hit enter. If no error is thrown, you are ready to rock.

If you did get an error, or run into other installation issues, please let us know or better yet post on Stack Overflow (with the healthcare-ai tag) so we can help others along this process.

Getting started

  1. Read through the Getting Started section of the healthcareai-py documentation.

  2. Read through the example files to learn how to use the healthcareai-py API.

    • For examples of how to train and evaluate a supervised model, inspect and run either or using our sample diabetes dataset.
    • For examples of how to use a model to make predictions, inspect and run either or after running one of the first examples.
    • For examples of more advanced use cases, inspect and run
  3. To train and evaluate your own model, modify the queries and parameters in either or to match your own data.

  4. Decide what type of prediction output you want. See Choosing a Prediction Output Type for details.

  5. Set up your database tables to match the schema of the output type you chose.

  6. Congratulations! After running one of the example files with your own data, you should have a trained model. To use your model to make predictions, modify either or to use your new model. You can then run it to see the results.

For Issues