plot_famTEMsel: plot component functions from a 'famTEMsel' object

In syhyunpark/famTEMsel: Functional Additive Models for Treatment Effect-Modifier Selection

Description

Produces plots of the component functions and the single-index coefficient functions from a famTEMsel object.

Usage

plot_famTEMsel(famTEMsel.obj, newX = NULL, newZ = NULL, newA = NULL,
  scatter.plot = TRUE, lambda.index = famTEMsel.obj$lambda.index,
  which.index = famTEMsel.obj$nonzero.index, ylims,
  single.index.plot = TRUE, solution.path = FALSE)

Arguments

famTEMsel.obj	a famTEMsel object
newX	a (n by p) list of new values for the functional covariates X at which predictions are to be made; the jth element of the list corresponds to a n-by-n.eval[j] matrix of the observed jth functional covariates; n.eval[j] represents the number of evaluation points of the jth functional covariates; if NULL, X from the training set is used.
newZ	a (n by q) matrix of new values for the scalar covariates Z at which predictions are to be made; if NULL, Z from the training set is used.
newA	a (n-by-1) vector of new values for the treatment A at which plots are to be made; the default is NULL, in which case A is taken from the training set.
scatter.plot	if TRUE, draw scatter plots of partial residuals versus the covariates; these scatter plots are made based on the training observations; the default is TRUE.
lambda.index	an index of the tuning parameter lambda at which plots are to be made; one can supply lambda.opt.index obtained from the function cv.samTEMsel; the default is NULL, in which case plot_samTEMsel utilizes the most non-sparse model.
which.index	this specifies which component functions are to be plotted; the default is all p component functions, i.e., 1:p.
ylims	this specifies the vertical range of the plots, e.g., c(-10, 10).
single.index.plot	if TRUE, draw the plots of the estimated single-index coefficient functions; the default is TRUE.
solution.path	if TRUE, draw the functional norms of the fitted component functions (based on the training set) versus the regularization parameter; the default is FALSE.

Author(s)

Park, Petkova, Tarpey, Ogden

See Also

famTEMsel,predict_famTEMsel, cv.famTEMsel

syhyunpark/famTEMsel documentation built on Feb. 6, 2020, 12:13 a.m.