roc 0.0.0 → 1.0.0

data/lib/roc.rb

@@ -49,6 +49,7 @@ class ROCarray < Array
     ptot = self.join.count('P')
     return nil if ptot==0
     ntot = self.join.count('N')
+    return nil if ntot==0
     ystep = 1.0 / ptot
     xstep = 1.0 / ntot
     parr = self.inject([[0,0]]) do |points,rocpoint|

data/roc.gemspec

@@ -4,7 +4,7 @@
 Gem::Specification::new do |spec|
   spec.name = "roc"
   spec.description = 'Receiver operator characteristic (ROC)' 
-  spec.version = "0.0.0"
+  spec.version = "1.0.0"
   spec.platform = Gem::Platform::RUBY
   spec.summary = "roc"
 

data/samples/a.rb

@@ -7,3 +7,6 @@
   r1.roc
   r1.auc
   r1.auc(0.1)
+  r2.roc
+  r2.auc
+  r2.auc(0.1)

data/test/roc.rb

@@ -4,51 +4,41 @@ $:.unshift('..')
 $:.unshift('../lib')
 
 require('test/unit')
-require('forkoff')
+require('roc')
 
 
 class T < Test::Unit::TestCase
 
-# simple usage
-#
-
+  # simple usage
   def test_0010
-    results = [0,1,2].forkoff!{|n| [n, Process.pid]}
-    assert_equal(results, results.uniq, 'all ran in a different process')
+    r = ROCarray.new %w(P N P) 
+    assert r.auc == 0.5
   end
 
-# it's faster
-#
+  # without positive samples, it should be nil
   def test_0020
-    n = 4
-    strategies = :forkoff, :each
-
-    4.times do
-      result = {}
-
-      strategies.each do |strategy|
-        a = Time.now.to_f
-        (0..4).send(strategy){|i| sleep 0.2}
-        b = Time.now.to_f
-        elapsed = b - a
-        result[strategy] = elapsed
-      end
-
-      assert result[:forkoff] < result[:each], 'forkoff is faster than each for long running tasks'
-    end
+    r = ROCarray.new %w(NN)
+    assert r.auc == nil
   end
 
-# in case of different execution times for different processes
-#
-  def test_0030
-    a = Time.now.to_f
-    (0...4).forkoff(2) do |i|
-      sleep i.modulo(2)
-    end
-    b = Time.now.to_f
-    elapsed = b - a
-    assert elapsed < 2
+  # without negative samples, it should be nil
+  def test_0025
+    r = ROCarray.new %w(PP)
+    assert r.auc == nil
   end
 
+  # only equal predictions -> 0.5
+  def test_0030
+    r = ROCarray.new %w(PPPPPPPPPPNNNNNNNNNN)
+    d = 1e-12
+    assert r.auc>0.5-d and r.auc<0.5+d
+  end
+  
+  # initialization with prediction values
+  def test_0040
+    assert ROCarray.new([1.0 , 2.0 , 3.0],[0]).auc==0.0 and
+           ROCarray.new([1.0 , 2.0 , 3.0],[1]).auc==0.5 and
+           ROCarray.new([1.0 , 2.0 , 3.0],[2]).auc==1.0 
+  end
 
 end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.4
 specification_version: 1
 name: roc
 version: !ruby/object:Gem::Version 
-  version: 0.0.0
-date: 2009-10-21 00:00:00 +09:00
+  version: 1.0.0
+date: 2009-11-28 00:00:00 +09:00
 summary: roc
 require_paths: 
 - lib