cascade.nn.SpecNorm

class cascade.nn.SpecNorm(n_vars, n_particles, n_iter=5)[source]

Bases: AcycPrior

Spectral norm penalized log prior probability

Parameters:

  • n_vars (int) – Number of variables in the graph

  • n_particles (int) – Number of SVGD particles

  • n_iter (int) – Number of power iterations

Methods

compute_limit

Detection limit on the complete DAG

dot

Vector dot product with particles in the last dimension

energy

Spectral norm acyclic energy function (negative log probability) of a graph scaffold

mv

Sparse matrix-vector product with particles in the last dimension

power_iteration