healthcareai-r
healthcareai-r copied to clipboard
R tools for healthcare machine learning
healthcareai ![](https://github.com/HealthCatalyst/healthcareai-r/raw/main/man/figures/logo.png)
Overview
The aim of healthcareai
is to make machine learning in healthcare as
easy as possible. It does that by providing functions to:
- Develop customized, reliable, high-performance machine learning models with minimal code
- Easily make and evaluate predictions and push them to a database
- Understand how a model makes its predictions
- Make data cleaning, manipulation, imputation, and visualization as simple as possible
Usage
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, eXtreme Gradient Boosting, and glmnet
# > Model Name: diabetes
# > Target: diabetes
# > Class: Classification
# > Performance Metric: AUROC
# > Number of Observations: 768
# > Number of Features: 12
# > Models Trained: 2018-09-01 18:19:44
# >
# > 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 = 12
Make predictions and examine predictive performance:
predictions <- predict(models, outcome_groups = TRUE)
plot(predictions)
Learn More
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
?machine_learn
, ?predict.model_list
, and ?explore
.
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.
Get Involved
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.
Contributing
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.
Feedback
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, devtools::session_info()
.
Legacy
Version 1 of healthcareai
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:
install.packages("remotes"); remotes::install_github("HealthCatalyst/[email protected]")
.
For an example of how to adapt v1 models to the v2 API, check out the Transitioning vignettes.