The aim of
healthcareai is to make machine learning in healthcare as easy as possible. It does that by providing functions to:
healthcareai can take you from messy data to an optimized model in one line of code:
models <- machine_learn(pima_diabetes, patient_id, outcome = diabetes) models # > Algorithms Trained: Random Forest, k-Nearest Neighbors # > Target: diabetes # > Class: Classification # > Performance Metric: AUROC # > Number of Observations: 768 # > Number of Features: 12 # > Models Trained: 2018-04-20 12:25:41 # > # > Models tuned via 5-fold cross validation over 10 combinations of hyperparameter values. # > Best model: Random Forest # > AUPR = 0.71, AUROC = 0.84 # > Optimal hyperparameter values: # > mtry = 2 # > splitrule = extratrees # > min.node.size = 3
Make predictions and examine predictive performance:
predictions <- predict(models) plot(predictions)
For details on what’s happening under the hood and for options to customize data preparation and model training, see Getting Started with healthcareai as well as the helpfiles for individual functions such as
Documentation of all functions as well as vignettes on various uses of the package are available at the package website: https://docs.healthcare.ai/.
Also, be sure to read our blog and watch our broadcasts to learn more about what’s new in healthcare machine learning and how we are using this toolkit to put machine learning to work in real healthcare systems.
We have a Slack community that is a great place to introduce yourself, share what you’re doing with the package, ask questions, and troubleshoot your code.
If you are interested in contributing the package (great!), please read the contributing guide, and look for issues with the “help wanted” tag. Feel free to tackle any issue that interests you; those are a few issues that we feel would make a good place to start.
Your feedback is hugely appreciated. It is makes the package work well and helps us make it more useful to the community.
Both feature requests and bug reports should be submitted as Github issues.
Bug reports should be filed with a minimal reproducable example. The reprex package is extraordinarily helpful for this. Please also include the output of
sessionInfo() or better yet,
Version 1 of healthcare-ai has been retired. You can continue to use it, but its compatibility with changes in the R ecosystem are not guaranteed. You should always be able to install it from github with:
For an example of how to adapt v1 models to the v2 API, check out the Transitioning vignettes.