Samplers¶
-
class
geoopt.samplers.
RHMC
(params, epsilon=0.001, n_steps=1)[source]¶ Riemannian Hamiltonian Monte-Carlo.
Parameters:
-
class
geoopt.samplers.
RSGLD
(params, epsilon=0.001)[source]¶ Riemannian Stochastic Gradient Langevin Dynamics.
Parameters: - params (iterable) – iterables of tensors for which to perform sampling
- epsilon (float) – step size