bayespecon.SARTobit¶
-
class bayespecon.SARTobit(*args, censoring=
0.0, **kwargs)[source]¶ Bayesian spatial autoregressive Tobit model.
\[y^* = \rho W y^* + X\beta + \varepsilon,\quad \varepsilon \sim N(0,\sigma^2 I),\]with observed outcome
\[y = \max(c, y^*)\]where
cis the left-censoring point (default0). Censored observations contribute their CDF to the likelihood; uncensored observations contribute the density of \(y^*\) evaluated at \(y\).- Parameters:¶
- formula : str, optional
Wilkinson-style formula, e.g.
"y ~ x1 + x2". Requiresdata.- data : pandas.DataFrame or geopandas.GeoDataFrame, optional
Data source for formula mode.
- y : array-like, optional
Observed (censored) response of shape
(n,). Required in matrix mode.- X : array-like or pandas.DataFrame, optional
Design matrix. Required in matrix mode.
- W : libpysal.graph.Graph or scipy.sparse matrix
Spatial weights of shape
(n, n); seeSARfor accepted formats.- censoring : float, default 0.0¶
Left-censoring threshold
c. Observations withy <= censoringare treated as censored and the latent \(y^*\) is sampled.- priors : dict, optional
Override default priors. Supported keys:
rho_lower(float, default -1.0): Lower bound of Uniform prior on \(\rho\).rho_upper(float, default 1.0): Upper bound of Uniform prior on \(\rho\).beta_mu(float, default 0.0): Normal prior mean for \(\beta\).beta_sigma(float, default 1e6): Normal prior std for \(\beta\).sigma_sigma(float, default 10.0): HalfNormal prior std for \(\sigma\).censor_sigma(float, default 10.0): HalfNormal scale for the latenty_cens_gapshifting censored draws belowc.nu_lam(float, default 1/30): Rate of TruncExp(lower=2) prior on \(\nu\) (only used whenrobust=True).
- logdet_method : str, optional
How to compute \(\log|I - \rho W|\).
None(default) auto-selects"eigenvalue"forn <= 2000else"chebyshev".- robust : bool, default False
If True, replace the Normal innovation with Student-t. See Robust regression below.
Notes
Robust regression
When
robust=True, the error distribution is changed from Normal to Student-t. For uncensored observations the density becomes:\[f(y^*_i \mid \mu_i, \sigma, \nu) = \frac{1}{\sigma} \, t_\nu\!\left(\frac{y^*_i - \mu_i}{\sigma}\right)\]and for censored observations the probability becomes:
\[P(y^*_i \le c) = T_\nu\!\left(\frac{c - \mu_i}{\sigma}\right)\]where \(T_\nu\) is the Student-t CDF with \(\nu\) degrees of freedom, and \(\nu \sim \mathrm{TruncExp}(\lambda_\nu, \mathrm{lower}=2)\) with rate
nu_lam(default 1/30). The defaultnu_lam = 1/30gives a prior mean of approximately 30.Methods
__init__(*args[, censoring])fit([draws, tune, chains, target_accept, ...])Sample posterior and attach pointwise log-likelihood for IC metrics.
Return fitted values at posterior mean parameters.
Return residuals on the observed scale.
Run Bayesian LM specification tests and return a summary table.
spatial_diagnostics_decision([alpha, format])Return a model-selection decision from Bayesian LM test results.
spatial_effects([return_posterior_samples])Compute Bayesian inference for direct, indirect, and total impacts.
summary([var_names])Return posterior summary table.
Attributes
Return the ArviZ InferenceData from the most recent fit.
Return the PyMC model object built for the most recent fit.
-
fit(draws=
2000, tune=1000, chains=4, target_accept=0.9, random_seed=None, idata_kwargs=None, **sample_kwargs)[source]¶ Sample posterior and attach pointwise log-likelihood for IC metrics.
The SAR Tobit model uses
pm.Potentialfor both the residual log-likelihood and the Jacobian, so nothing is auto-captured. We compute the complete pointwise log-likelihood manually after sampling, using the Tobit censoring formula:Uncensored: log N(y | mu, sigma^2)
Censored: log Phi((c - mu) / sigma)
- property inference_data : arviz.data.inference_data.InferenceData | None[source]¶
Return the ArviZ InferenceData from the most recent fit.
- property pymc_model : pymc.model.core.Model | None[source]¶
Return the PyMC model object built for the most recent fit.
For Gibbs-fitted models the PyMC model is not constructed during sampling; it is built lazily on first access so that downstream consumers (e.g. bridge sampling for marginal likelihoods) can evaluate
logpand the prior under the same model definition used by the NUTS path.
- spatial_diagnostics()[source]¶
Run Bayesian LM specification tests and return a summary table.
Looks up the diagnostic suite registered for this model class and calls each test function on this fitted model, collecting the results into a tidy DataFrame. The set of tests depends on the model type — for example, an OLS model runs LM-Lag, LM-Error, LM-SDM-Joint, and LM-SLX-Error-Joint, while an SAR model runs LM-Error, LM-WX, and Robust-LM-WX.
Requires the model to have been fit (
.fit()called) and a spatial weights matrixWto have been supplied at construction time.- Returns:¶
DataFrame indexed by test name with columns:
Column
Description
statistic
Posterior mean of the LM statistic
median
Posterior median of the LM statistic
df
Degrees of freedom for the \(\chi^2\) reference
p_value
Bayesian p-value:
1 - chi2.cdf(mean, df)ci_lower
Lower bound of 95% credible interval (2.5%)
ci_upper
Upper bound of 95% credible interval (97.5%)
The DataFrame has
attrs["model_type"](class name) andattrs["n_draws"](total posterior draws) metadata.- Return type:¶
pandas.DataFrame
- Raises:¶
RuntimeError – If the model has not been fit yet.
ValueError – If no spatial weights matrix
Wwas supplied.
See also
spatial_diagnostics_decisionModel-selection decision based on the test results.
spatial_effectsPosterior inference for direct/indirect/total impacts.
Examples
>>> ols = OLS(formula="price ~ income + crime", data=df, W=w) >>> ols.fit() >>> ols.spatial_diagnostics() statistic median df p_value ci_lower ci_upper LM-Lag 3.21 2.98 1 0.073 0.12 8.54 LM-Error 5.67 5.34 1 0.017 0.34 12.10 LM-SDM-Joint 7.89 7.12 4 0.096 1.23 18.32 LM-SLX-Error-Joint 6.45 5.98 4 0.168 0.89 15.67
-
spatial_diagnostics_decision(alpha=
0.05, format='graphviz')[source]¶ Return a model-selection decision from Bayesian LM test results.
Implements the decision tree from Koley and Bera [2024] (the Bayesian analogue of the classical
stge_kbprocedure in Anselin et al. [1996]). The decision logic depends on the current model type and the pattern of significant tests:From OLS (6-test decision tree):
If only LM-Lag is significant → SAR.
If only LM-Error is significant → SEM.
If both are significant → use the Anselin–Florax / Koley–Bera robust pair: Robust-LM-Lag → SAR, Robust-LM-Error → SEM, both → SARAR. If neither robust test is significant, fall back to the lower raw p-value.
If neither naive test is significant → OLS.
From SAR (3-test decision tree):
LM-Error significant → SARAR; LM-WX significant → SDM; Robust-LM-WX significant → SDM.
From SEM (2-test decision tree):
LM-Lag significant → SARAR; LM-WX significant → SDEM.
From SLX (4-test decision tree):
Robust-LM-Lag-SDM significant → SDM; Robust-LM-Error-SDEM significant → SDEM; both → MANSAR; neither → SLX.
From SDM: LM-Error-SDM significant → MANSAR; else SDM.
From SDEM: LM-Lag-SDEM significant → MANSAR; else SDEM.
- Parameters:¶
- alpha : float, default 0.05¶
Significance level for the Bayesian p-values.
- format : {"graphviz", "ascii", "model"}, default "graphviz"¶
Output format.
"model"returns the recommended-model name string."ascii"returns an indented box-drawing rendering of the full decision tree with the chosen path highlighted."graphviz"returns agraphviz.Digraphobject that renders inline in Jupyter; if the optionalgraphvizpackage is not installed aUserWarningis issued and the ASCII rendering is returned instead.
- Returns:¶
Recommended model name when
format="model", an ASCII tree string whenformat="ascii", or agraphviz.Digraphwhenformat="graphviz"(with ASCII fallback on missing dep).- Return type:¶
str or graphviz.Digraph
See also
spatial_diagnosticsCompute the Bayesian LM test statistics.
References
-
spatial_effects(return_posterior_samples=
False)[source]¶ Compute Bayesian inference for direct, indirect, and total impacts.
Computes impact measures for each posterior draw, then summarises the posterior distribution with means, 95% credible intervals, and Bayesian p-values. This is the fully Bayesian analog of the simulation-based approach in LeSage and Pace [2009] and the asymptotic variance formulas in Arbia et al. [2020].
Models without a spatial lag on y do not exhibit global feedback propagation through \((I-\rho W)^{-1}\). However, models with spatially lagged covariates (SLX, SDEM) can still have non-zero neighbour spillovers captured in the indirect term.
- Parameters:¶
- Returns:¶
If return_posterior_samples is
False(default), returns a DataFrame indexed by feature names with columns for posterior means, credible-interval bounds, and Bayesian p-values.If return_posterior_samples is
True, returns(DataFrame, dict)where the dict has keys"direct","indirect","total", each mapping to a(G, k)array of posterior draws.- Return type:¶