test_sgld.py 文件源码

def test_sgld_dense(self):
        tf.reset_default_graph()

        x = tf.Variable(tf.zeros(20), dtype=tf.float32)
        loss = tf.reduce_sum(tf.square(x - 10))

        sgld = SGLD(learning_rate=0.4)
        train_op_sgld = sgld.minimize(loss)

        sess = tf.InteractiveSession()
        sess.run(tf.global_variables_initializer())

        sess.run(train_op_sgld)
        xh = sess.run(x)
        self.assertTrue(5.0 <= xh.mean() and xh.mean() <= 11.0)