estimate: Extract Posterior Potential Outcomes

In PStrata: Principal Stratification Analysis in R

Extract Posterior Potential Outcomes

Description

Maps posterior samples of outcome-group means back to the (stratum x treatment) grid via the SZDG table.

Usage

estimate(object, ...)

## S3 method for class 'PStrataFit'
estimate(object, type = c("probability", "RACE"), ...)

Arguments

object	A PStrataFit object.
...	Additional arguments (unused).
type	For survival models: "probability" (survival probability) or "RACE" (restricted average causal effect).

Value

A PSEstimate object containing

outcome_array	[S, Z, iter] array for non-survival, or [S, Z, T, iter] for survival.
is_survival	Logical.
time_points	Numeric vector (survival only).
model_info	List of strata names, treatment info, family, ER flags, outcome groups.

Examples

data(sim_data_normal)
model <- PStrataModel(
  S.formula = Z + D ~ 1,
  Y.formula = Y ~ 1,
  Y.family  = gaussian(),
  strata    = c(n = "00", c = "01", a = "11"),
  ER        = c("n", "a")
)
ps_fit <- fit(model, data = sim_data_normal, chains = 2, iter = 500)
est <- estimate(ps_fit)
summary(est)
plot(est)

PStrata documentation built on May 14, 2026, 5:06 p.m.