bayespecon.models.flow_panel.PoissonSARFlowPanel

class bayespecon.models.flow_panel.PoissonSARFlowPanel(y, G, X, **kwargs)[source]

Panel Poisson spatial-lag flow model with unrestricted dependence.

The stacked panel counts satisfy

\[y_{ij,t} \sim \operatorname{Poisson}(\lambda_{ij,t}), \qquad \log \boldsymbol{\lambda}_t = A(\rho_d, \rho_o, \rho_w)^{-1} X_t \beta,\]

where

\[A(\rho_d, \rho_o, \rho_w) = I_N - \rho_d W_d - \rho_o W_o - \rho_w W_w.\]
Parameters:
y : array-like of int

Stacked panel flow counts in shape (T, n, n), (T, n^2), or (n^2 * T,). Non-integer values raise an error if they do not round exactly.

G : libpysal.graph.Graph

Row-standardised graph on n units.

X

Stacked panel design matrix in time-first order.

T : int

Number of panel periods (must be a positive integer).

col_names : list of str, optional

Feature names for X. Inferred from a DataFrame if omitted.

k : int, optional

Number of destination/origin covariate pairs used by flow effects; inferred from columns prefixed dest_ if omitted.

model : int, default 0

Fixed-effects transform. Only 0 (pooled) is supported — within transforms break the non-negative integer support.

logdet_method : str, default "traces"

Log-determinant method. Only "traces" is supported here.

restrict_positive : bool, default True

If True, use pm.Dirichlet("rho_simplex", a=ones(4)) to enforce \(\rho_d, \rho_o, \rho_w \geq 0\) and \(\rho_d + \rho_o + \rho_w \leq 1\). If False, three independent pm.Uniform(rho_lower, rho_upper) priors are used with a differentiable quadratic-wall stability potential.

miter : int, default 30

Trace polynomial order for the log-determinant.

titer : int, default 800

Geometric tail cutoff for the log-determinant series.

trace_riter : int, default 50

Number of Monte Carlo probes for trace estimation.

trace_seed : int, optional

Random seed for trace estimation reproducibility.

symmetric_xo_xd : bool, optional

If None (default), origin and destination design blocks are compared and symmetry is auto-detected.

priors : dict, optional

Override default priors. Supported keys:

  • beta_mu : float, default 0.0 — Normal prior mean for beta.

  • beta_sigma : float, default 10.0 — Normal prior std for beta.

  • rho_lower : float, default -1.0 — Lower bound of Uniform prior on each ρ (only when restrict_positive=False).

  • rho_upper : float, default 1.0 — Upper bound of Uniform prior on each ρ (only when restrict_positive=False).

Notes

This class currently supports pooled panels only (model=0). Within transforms are not valid for Poisson counts because they break the non-negative integer support. There is no sigma parameter and robust (Student-t) likelihoods are not supported.

__init__(y, G, X, **kwargs)[source]

Methods

__init__(y, G, X, **kwargs)

fit([draws, tune, chains, target_accept, ...])

Draw samples from the posterior.

fit_approx([draws, n, method, random_seed, ...])

Fit a variational approximation and return posterior draws.

posterior_predictive([n_draws, random_seed, ...])

Draw posterior-predictive samples y_rep for the full panel stack.

spatial_diagnostics()

Run Bayesian LM specification tests for flow panel models.

spatial_diagnostics_decision([alpha, ...])

Return a model-selection decision from Bayesian LM test results.

spatial_effects([draws, ...])

Summarise posterior origin/destination/intra/network/total effects.

summary([var_names])

Return posterior summary table via ArviZ.

Attributes

approximation

Return the most recent PyMC variational approximation, if any.

inference_data

Return posterior draws from the most recent fit.

pymc_model

Return the most recently built PyMC model.

property approximation[source]

Return the most recent PyMC variational approximation, if any.

fit(draws=2000, tune=1000, chains=4, target_accept=0.9, random_seed=None, store_lambda=False, idata_kwargs=None, **sample_kwargs)[source]

Draw samples from the posterior.

fit_approx(draws=2000, n=10000, method='advi', random_seed=None, store_lambda=False, compute_log_likelihood=True, **fit_kwargs)[source]

Fit a variational approximation and return posterior draws.

property inference_data : arviz.data.inference_data.InferenceData | None[source]

Return posterior draws from the most recent fit.

posterior_predictive(n_draws=None, random_seed=None, parallel=-1)[source]

Draw posterior-predictive samples y_rep for the full panel stack.

Parameters:
n_draws : int, optional

Number of posterior draws to use. Defaults to all.

random_seed : int, optional

Seed for the noise/Poisson sampler.

parallel : int or None, default -1

Number of worker threads for the per-draw loop. -1 uses os.cpu_count(); None/0/1 forces sequential execution. Reproducibility under a fixed random_seed is preserved across worker counts via SeedSequence.spawn.

Returns:

Array of shape (n_draws, N_flow * T) with posterior-predictive flows in time-first stacked order.

Return type:

np.ndarray

property pymc_model : pymc.model.core.Model | None[source]

Return the most recently built PyMC model.

spatial_diagnostics()[source]

Run Bayesian LM specification tests for flow panel models.

See bayespecon.models.base.SpatialModel.spatial_diagnostics() for the column schema.

Raises:

RuntimeError – If the model has not been fit yet.

spatial_diagnostics_decision(alpha=0.05, format='graphviz', theme='default')[source]

Return a model-selection decision from Bayesian LM test results.

Walks the panel-flow decision tree using Bayesian p-values from spatial_diagnostics() and recommends either OLSFlowPanel (no spatial dependence detected) or SARFlowPanel (at least one direction is significant).

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 tree. "graphviz" returns a graphviz.Digraph (with ASCII fallback if graphviz is not installed).

Return type:

str or graphviz.Digraph

spatial_effects(draws=None, return_posterior_samples=False, ci=0.95, mode='auto', parallel=-1)[source]

Summarise posterior origin/destination/intra/network/total effects.

See bayespecon.models.flow.FlowModel.spatial_effects() for the mode semantics (auto / combined / separate destination-origin sides per Thomas-Agnan & LeSage 2014, §83.5.2) and the parallel kwarg.

summary(var_names=None, **kwargs)[source]

Return posterior summary table via ArviZ.