Example #1
0
    def testSigmoidCrossEntropyOneHot(self):
        logits = np.array([[-1., 1.], [1., -1.], [1., -1.], [1., -1.]])
        labels = np.array([[0, 1], [1, 0], [0, 0], [0, 1]])
        logits = np.expand_dims(np.expand_dims(logits, 1), 1)
        labels = np.expand_dims(np.expand_dims(labels, 1), 1)

        with self.test_session() as session:
            score, _ = metrics.sigmoid_cross_entropy_one_hot(logits, labels)
            session.run(tf.global_variables_initializer())
            session.run(tf.local_variables_initializer())
            s = session.run(score)
        self.assertAlmostEqual(s, 0.688, places=3)
Example #2
0
  def testSigmoidCrossEntropyOneHot(self):
    logits = np.array([
        [-1., 1.],
        [1., -1.],
        [1., -1.],
        [1., -1.]
    ])
    labels = np.array([
        [0, 1],
        [1, 0],
        [0, 0],
        [0, 1]
    ])
    logits = np.expand_dims(np.expand_dims(logits, 1), 1)
    labels = np.expand_dims(np.expand_dims(labels, 1), 1)

    with self.test_session() as session:
      score, _ = metrics.sigmoid_cross_entropy_one_hot(logits, labels)
      session.run(tf.global_variables_initializer())
      session.run(tf.local_variables_initializer())
      s = session.run(score)
    self.assertAlmostEqual(s, 0.688, places=3)