Python vq_nearest_neighbor示例

示例#1

 def testVQNearestNeighbors(self):
   x = tf.constant([[0, 0.9, 0], [0.8, 0., 0.]], dtype=tf.float32)
   means = tf.constant(
       [[1, 0, 0], [0, 1, 0], [0, 0, 1], [9, 9, 9]], dtype=tf.float32)
   x_means_hot, _, _ = discretization.vq_nearest_neighbor(x, means)
   x_means_hot_test = np.array([[0, 1, 0, 0], [1, 0, 0, 0]])
   x_means_hot_eval = self.evaluate(x_means_hot)
   self.assertEqual(np.shape(x_means_hot_eval), (2, 4))
   self.assertTrue(np.all(x_means_hot_eval == x_means_hot_test))

示例#2

文件： discretization_test.py 项目： qixiuai/tensor2tensor

 def testVQNearestNeighbors(self):
   x = tf.constant([[0, 0.9, 0], [0.8, 0., 0.]], dtype=tf.float32)
   means = tf.constant(
       [[1, 0, 0], [0, 1, 0], [0, 0, 1], [9, 9, 9]], dtype=tf.float32)
   x_means_hot, _, _ = discretization.vq_nearest_neighbor(x, means)
   x_means_hot_test = np.array([[0, 1, 0, 0], [1, 0, 0, 0]])
   x_means_hot_eval = self.evaluate(x_means_hot)
   self.assertEqual(np.shape(x_means_hot_eval), (2, 4))
   self.assertTrue(np.all(x_means_hot_eval == x_means_hot_test))

示例#3

文件： discretization_test.py 项目： jaywalnut310/tensor2tensor

 def testVQNearestNeighbors(self):
     x = tf.constant([[0, 0.9, 0], [0.8, 0., 0.]], dtype=tf.float32)
     means = tf.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1], [9, 9, 9]],
                         dtype=tf.float32)
     x_means_hot, _ = discretization.vq_nearest_neighbor(x, means)
     x_means_hot_test = np.array([[0, 1, 0, 0], [1, 0, 0, 0]])
     with self.test_session() as sess:
         tf.global_variables_initializer().run()
         x_means_hot_eval = sess.run(x_means_hot)
         self.assertEqual(np.shape(x_means_hot_eval), (2, 4))
         self.assertTrue(np.all(x_means_hot_eval == x_means_hot_test))

示例#4

文件： discretization_test.py 项目： kltony/tensor2tensor

 def testVQNearestNeighbors(self):
   x = tf.constant([[0, 0.9, 0], [0.8, 0., 0.]], dtype=tf.float32)
   means = tf.constant(
       [[1, 0, 0], [0, 1, 0], [0, 0, 1], [9, 9, 9]], dtype=tf.float32)
   x_means_hot, _ = discretization.vq_nearest_neighbor(x, means)
   x_means_hot_test = np.array([[0, 1, 0, 0], [1, 0, 0, 0]])
   with self.test_session() as sess:
     tf.global_variables_initializer().run()
     x_means_hot_eval = sess.run(x_means_hot)
     self.assertEqual(np.shape(x_means_hot_eval), (2, 4))
     self.assertTrue(np.all(x_means_hot_eval == x_means_hot_test))