Exemplo n.º 1
0
    def testSigmoidAccuracyOneHot(self):
        logits = np.array([[-1., 1.], [1., -1.], [-1., 1.], [1., -1.]])
        labels = np.array([[0, 1], [1, 0], [1, 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_accuracy_one_hot(logits, labels)
            session.run(tf.global_variables_initializer())
            session.run(tf.local_variables_initializer())
            s = session.run(score)
        self.assertEqual(s, 0.5)
Exemplo n.º 2
0
  def testSigmoidAccuracyOneHot(self):
    logits = np.array([
        [-1., 1.],
        [1., -1.],
        [-1., 1.],
        [1., -1.]
    ])
    labels = np.array([
        [0, 1],
        [1, 0],
        [1, 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_accuracy_one_hot(logits, labels)
      session.run(tf.global_variables_initializer())
      session.run(tf.local_variables_initializer())
      s = session.run(score)
    self.assertEqual(s, 0.5)