cascade.nn.Prior

class cascade.nn.Prior(n_vars, n_particles, **kwargs)[source]

Bases: Module

Compute unnormalized negative log prior probability of a scaffold graph

Parameters:

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

  • n_particles (int) – Number of SVGD particles

Methods

energy

Energy function (negative log probability) of a graph scaffold