Software

Research software maintained by Ruoqing Zhu and collaborators.

Research software for tree-based learning, survival analysis, dimension reduction, and related methods. Source code is also available through my GitHub profile.

RLT

CRANv3.2.6 GitHubv6.1.1

Reinforcement Learning Trees provides tree-based learning tools, including an implemented survival analysis component. Some features remain experimental.

Example file CRAN Task View

Additional packages

orthoDr

CRANv0.6.8 GitHubv2.0.0

Orthogonality Constrained Optimization and Dimension Reduction

This package offers an optimization solver for orthogonality constrained problems, $$\min f(X) \quad \text{subject to} \quad X^T X = I,$$ using the algorithm developed by Wen and Yin (2013). We use it for semiparametric dimension reduction and personalized medicine problems.

Examples

drforest

Dimension Reduction Forests

This package provides estimation methods for dimension reduction forests and local subspace variable importance. Local subspace variable importance can help detect influential variables in personalized prediction and recommendation.

GitHub repository

MOTE.RF

Random Forests for Heterogeneous Treatment Effect Estimation with Multiple Responses

Designed for settings in which covariates $X$ and multivariate response $Y$ are both measured before and after an intervention. It has been applied to a microbiome study to estimate dietary effects on multiple health outcomes.

GitHub repository

MatrixFact

A Nonnegative Matrix Factorization Toolbox

This package implements NMF, ONMF, semi-NMF, semi-orthogonal NMF, and extensions to binary data. Semi-orthogonal NMF can be applied to document-word matrices for text analysis.

GitHub repository

RIST

Recursively Imputed Survival Trees

The R package is planned for integration with RLT. The current implementation and example files are available below.

Example Main function Supporting functions