roc 0.0.0

data/README

@@ -0,0 +1,25 @@
+NAME
+
+  roc
+
+SYNOPSIS
+
+  Receiver operator characteristic
+
+URI
+
+  http://rubyforge.org/projects/roc
+  http://github.com/fredrikj/roc
+
+INSTALL
+
+  gem install roc
+
+DESCRIPTION
+
+  Calculates a ROC curve as well as the area under the curve (AUC).
+
+HISTORY
+  0.0.0
+
+    initial version

data/lib/roc.rb

@@ -0,0 +1,83 @@
+#----------
+# ROCarray
+#----------
+# An Array consisting only
+# of 'P' and 'N' (for true positive and negative).
+# It is further assumed that self is an array of
+# predictions ordered such that most positive predictions
+# come first.
+# Finally - if some elements have been ranked equally, they
+# should appear together in the same element in the Array.
+class ROCarray < Array
+  # Two ways to initialize:
+  #
+  # 1) arg1 is prediction values
+  #    arg2 is indices of P elements in arg1
+  #
+  # 2) arg1 is an array of T and P: 
+  #    Example: %w(P PN P N N) 
+  #
+  def initialize(arg1,arg2=nil)
+    a = if arg2
+          pn = Array.new(arg1.size,'N')
+          arg2.each{|arg2i| pn[arg2i]='P'}
+          tmp = arg1.zip(pn).sort_by{|i,j| -i}
+          i=1
+          while tmp[i]
+            if tmp[i][0]==tmp[i-1][0]
+              tmp[i-1][1] += tmp[i][1]
+              tmp.delete_at i
+            else
+              i+=1
+            end
+          end
+          tmp.map{|i,j| j}
+        else
+          if arg1.find{|i| !i.is_a? String or i.upcase !~ /^[PN]+$/}
+            raise 'Incorrect Array for ROC' 
+          end
+          arg1.dup
+        end
+    a.map!{|i| i.upcase}
+    super(a)
+  end
+
+  # --- roc ---
+  # Returns an array of x and y values up until fpmax.
+  # Returns nil if there are no P values
+  def roc(fpmax=1)
+    ptot = self.join.count('P')
+    return nil if ptot==0
+    ntot = self.join.count('N')
+    ystep = 1.0 / ptot
+    xstep = 1.0 / ntot
+    parr = self.inject([[0,0]]) do |points,rocpoint|
+             break points if points.last[0]>=fpmax
+             curpoint = points.last.dup
+             rocpoint.split('').each do |char|
+               case char
+               when 'N' : curpoint[0]+=xstep
+               when 'P' : curpoint[1]+=ystep
+               end
+             end
+             points << curpoint
+           end
+    # Remove what is greater than fpmax
+    if parr.last[0]>fpmax
+      k = (parr[-1][1]-parr[-2][1]) / (parr[-1][0]-parr[-2][0])
+      r = fpmax - parr[-2][0]
+      parr[-1] = [ fpmax, parr[-2][1] + k * r ]
+    end
+    parr
+  end
+
+  # --- auc ---
+  def auc(fpmax=1)
+    r = self.roc(fpmax)
+    return nil unless r
+    (1...r.size).inject(0) do |area,i|
+      length =  [ r[i][0] - r[i-1][0], r[i][1] - r[i-1][1] ]
+      area += length[0] * ( r[i-1][1] + 0.5*length[1] )
+    end
+  end
+end

data/roc.gemspec

@@ -0,0 +1,25 @@
+## roc.gemspec
+#
+
+Gem::Specification::new do |spec|
+  spec.name = "roc"
+  spec.description = 'Receiver operator characteristic (ROC)' 
+  spec.version = "0.0.0"
+  spec.platform = Gem::Platform::RUBY
+  spec.summary = "roc"
+
+  spec.files = ["roc.gemspec", "lib", "lib/roc.rb", "README", "samples", "samples/a.rb", "test", "test/roc.rb"]
+  spec.executables = []
+  
+  spec.require_path = "lib"
+
+  spec.has_rdoc = true
+  spec.test_files = "test/roc.rb"
+
+  spec.extensions.push(*[])
+
+  spec.rubyforge_project = "roc"
+  spec.author = "Fredrik Johansson"
+  spec.email = "fredjoha@gmail.com"
+  spec.homepage = "http://github.com/fredrikj/roc/tree/master"
+end

data/samples/a.rb

@@ -0,0 +1,9 @@
+# A simple example of roc
+#
+  require 'roc'
+
+  r1 = ROCarray.new( %w(P N P) )
+  r2 = ROCarray.new( [0.3, 0.5, 0.7], [0, 2] )
+  r1.roc
+  r1.auc
+  r1.auc(0.1)

data/test/roc.rb

@@ -0,0 +1,54 @@
+$:.unshift('.')
+$:.unshift('./lib')
+$:.unshift('..')
+$:.unshift('../lib')
+
+require('test/unit')
+require('forkoff')
+
+
+class T < Test::Unit::TestCase
+
+# 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')
+  end
+
+# it's faster
+#
+  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
+  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
+  end
+
+
+end

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification 
+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
+summary: roc
+require_paths: 
+- lib
+email: fredjoha@gmail.com
+homepage: http://github.com/fredrikj/roc/tree/master
+rubyforge_project: roc
+description: Receiver operator characteristic (ROC)
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Fredrik Johansson
+files: 
+- roc.gemspec
+- lib
+- lib/roc.rb
+- README
+- samples
+- samples/a.rb
+- test
+- test/roc.rb
+test_files: 
+- test/roc.rb
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: []
+