fit: Fit a Principal Stratification Model

In PStrata: Principal Stratification Analysis in R

Fit a Principal Stratification Model

Description

Estimate a principal stratification model by running MCMC via Stan. The model specification (a PStrataModel) is combined with observed data to produce posterior samples.

Usage

fit(model, ...)

## S3 method for class 'PStrataModel'
fit(
  model,
  data,
  chains = 4,
  iter = 2000,
  warmup = floor(iter/2),
  cores = 1,
  seed = NULL,
  .debug = FALSE,
  ...
)

Arguments

model	A PStrataModel object.
...	Additional arguments passed to stan.
data	A data frame containing all variables referenced in the model.
chains, iter, warmup, cores, seed	MCMC settings passed to stan.
.debug	Logical. If TRUE, read Stan helper files from inst/ rather than installed package data. Useful during development.

Value

An object of class PStrataFit (or PStrataFitSurvival).

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)
summary(ps_fit)
diagnostics(ps_fit)
plot(ps_fit)
cat(stancode(ps_fit))

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