StratifiedMedicine (version 0.1.3)

ple_glmnet: Patient-level Estimates: Elastic Net (glmnet)

Description

Uses the elastic net (glmnet R package) to obtain patient-level estimates. Usable for continuous, binary, or survival outcomes.

Usage

ple_glmnet(Y, A, X, Xtest, lambda = "lambda.min", family, ...)

Arguments

Y

The outcome variable. Must be numeric or survival (ex; Surv(time,cens) )

A

Treatment variable. (a=1,...A)

X

Covariate space.

Xtest

Test set

lambda

Lambda for elastic-net (default = "lambda.min"). Other options include "lambda.1se" or fixed values

family

Outcome type ("gaussian", "binomial", "survival"), default is "gaussian"

...

Any additional parameters, not currently passed through.

Value

Trained glmnet model(s).

• mods - trained model(s)

• lambda - Lambda used for elastic-net (passes to prediction function)

• X - Covariate Space (in model matrix form)

Examples

# NOT RUN {
library(StratifiedMedicine)

## Continuous ##
dat_ctns = generate_subgrp_data(family="gaussian")
Y = dat_ctns$Y
X = dat_ctns$X
A = dat_ctns$A

mod1 = ple_glmnet(Y, A, X, Xtest=X, family="gaussian")
summary(mod1$mu_train$PLE)

# }