Project: hamiltonian-monte-carlo Author: matt-graham File: test_unconstrained.py (license) View Source Project

def test_init_with_energy_grad(self):
        mom_resample_coeff = 1.
        dtype = np.float64
        mass_matrix = np.array([[5., 3., -1], [3., 4., 2.], [-1., 2., 8.]])
        sampler = uhmc.EuclideanMetricHmcSampler(
            energy_func=energy_func,
            mass_matrix=mass_matrix,
            energy_grad=energy_grad,
            prng=self.prng,
            mom_resample_coeff=mom_resample_coeff,
            dtype=dtype)
        assert sampler.energy_func == energy_func
        assert np.allclose(sampler.mass_matrix, mass_matrix)
        assert np.allclose(
            sampler.mass_matrix_chol.dot(sampler.mass_matrix_chol.T),
            mass_matrix), 'Mass matrix Cholesky factor inconsistent.'
        assert sampler.energy_grad == energy_grad
        assert sampler.prng == self.prng
        assert sampler.mom_resample_coeff == mom_resample_coeff
        assert sampler.dtype == dtype
  • python

    Python (发音:[ paiθ(ə)n; (US) paiθɔn ]n. 蟒蛇,巨蛇 ),是一种面向对象的解释性的计算机程序设计语言,也是一种功能强大而完善的通用型语言,已经具有十多年的发…

    7952 引用 • 22 回帖 • 2 关注
  • numpy

    NumPy 是一个基础科学的计算包,包含: 一个强大的 N 维数组对象 sophisticated (broadcasting) functions tools for integrating C/C++ and Fortran code 有用的线性代数…

    1115 引用 • 1 关注
感谢    赞同    分享    收藏    关注    反对    举报    ...