Python AUCMetricの例

コード例 #1

ファイル: metrics_test.py プロジェクト: vcj-huy/lingvo

    def testAUCMetric(self):
        m = metrics.AUCMetric()
        m.Update(label=[1, 1], prob=[0.1, 0.2], weight=[1.0, 1.0])
        # No meaningful AUC yet, since all(labels==1) and function needs 2 types of
        # labels.
        self.assertEqual(0.0, m.value)

        m.Update(label=[0, 0], prob=[0.1, 0.2], weight=[1.0, 1.0])
        self.assertEqual(0.5, m.value)

コード例 #2

  def testAUCMetric(self):
    if not metrics.HAS_SKLEARN:
      self.skipTest('sklearn is not installed.')
    m = metrics.AUCMetric()
    m.Update(label=[1, 1], prob=[0.1, 0.2], weight=[1.0, 1.0])
    # No meaningful AUC yet, since all(labels==1) and function needs 2 types of
    # labels.
    self.assertEqual(0.0, m.value)

    m.Update(label=[0, 0], prob=[0.1, 0.2], weight=[1.0, 1.0])
    self.assertEqual(0.5, m.value)