Python sigmoid_precision_one_hotの例

コード例 #1

    def testSigmoidPrecisionOneHot(self):
        logits = np.array([[-1., 1.], [1., -1.], [1., -1.], [1., -1.]])
        labels = np.array([[0, 1], [0, 1], [0, 1], [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_precision_one_hot(logits, labels)
            session.run(tf.global_variables_initializer())
            session.run(tf.local_variables_initializer())
            s = session.run(score)
        self.assertEqual(s, 0.25)

コード例 #2

ファイル: metrics_test.py プロジェクト: kltony/tensor2tensor

  def testSigmoidPrecisionOneHot(self):
    logits = np.array([
        [-1., 1.],
        [1., -1.],
        [1., -1.],
        [1., -1.]
    ])
    labels = np.array([
        [0, 1],
        [0, 1],
        [0, 1],
        [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_precision_one_hot(logits, labels)
      session.run(tf.global_variables_initializer())
      session.run(tf.local_variables_initializer())
      s = session.run(score)
    self.assertEqual(s, 0.25)