test_capacities.py 文件源码

def test_tabular_UCB(self):
        nb_states = 3
        nb_actions = 2

        with tf.Graph().as_default():
            tf.set_random_seed(1)

            inputs_t = tf.random_uniform(shape=[1], minval=0, maxval=3, dtype=tf.int32)
            # inputs_t = tf.Print(inputs_t, data=[inputs_t], message='inputs_t')

            Qs = tf.ones([nb_states, nb_actions], dtype=tf.float32)
            # Qs = tf.Print(Qs, data=[Qs], message='Qs', summarize=12)

            actions_t, probs = capacities.tabular_UCB(Qs, inputs_t)
            # actions_t = tf.Print(actions_t, data=[timestep, actions_t], message='actions_t')

            with tf.Session() as sess:
                sess.run(tf.global_variables_initializer())

                inputs, actions = sess.run([inputs_t, actions_t])
                inputs, actions = sess.run([inputs_t, actions_t])
                inputs, actions = sess.run([inputs_t, actions_t])
                self.assertEqual(np.array_equal(inputs, [ 0 ]), True)
                self.assertEqual(np.array_equal(actions, [ 0 ]), True)