plot_ace: Plot predictions of the Additive Causal Expansion model

Description Usage Arguments Details Value

View source: R/plot_ace.R

Description

This function plots the response surface or the marginal effect fitted by the ace package around the training sample. For non-binary treatments, the method can display the interaction of the treatment with one confounder. The package plots:

Usage

1
2
3
plot_ace(object, Xcol, marginal = FALSE, plot3D = FALSE,
  show.observations = TRUE, Xlim = c(-1, 1), Xstep = 0.1, Zlim = c(-1,
  1), Zstep = 0.1, truefun, ...)

Arguments

object

An ace object as generated by the ace.train function.

Xcol

A numeric. If missing plots the 2D graph of Y on Z.

marginal

A logical statement, default: FALSE. If TRUE, plots the marginal surface of the response surface Y with respect to the basis expanded model Z using average X.

plot3D

A logical statement whether a three-dimensional plot should be produced. This is only valid if an X-column is selected and the plotly package is installed.

show.observations

A logical scalar that determines

Xlim

A two-dimensional vector with thelimits of the confounder dimension.

Xstep

A scalar for the step-size of the confounder dimension.

Zlim

A two-dimensional vector with thelimits of the treatment dimension.

Zstep

A scalar for the step-size of the treatment dimension.

truefun

A function handle for simulations.

...

Arguments handed to plotting function (plot, contour, plotly).

Details

Note: The method is using a colorblind-robust palette as in http://jfly.iam.u-tokyo.ac.jp/color/#what.

Value

Returns the plot object


mazphilip/GPspline documentation built on April 24, 2018, 3:26 a.m.