chao92 0.0.2 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0e80d9d9fb7fb96e66d48abcaf77e568241d69c8
-  data.tar.gz: cbd4f91ba21c8f581856921be847e6726b4c89c7
+  metadata.gz: a5a9b73405c74abb6de423ca788b1d0ef5279842
+  data.tar.gz: 1f63a237bfaa760a325f06176047b3800cd14a28
 SHA512:
-  metadata.gz: 45e404d9f67d57ed47749cd90f7adbb92a511267290858090faa7cead4377b7ae1a70e4c31512e3234a9ead42e9bdad6a221fb42335cd2fd7eb6e84b767f1433
-  data.tar.gz: acfeedd52b40352aaad3aede2b8161a4499943eeb098a1bd581222fcd01b086e0ab866e3a5d2a2b95415e15d2a0113d0ce4f352c08a3d406a922c1166a4f154e
+  metadata.gz: 73bb5c9186d5bd0d307b6953a0b5149dd5a2fb22ee5cc778441876c34a526e9e105acbc2cd1cf5437fbe5cf142c76f39763487d7de33e7dc9cffe9a072f5725e
+  data.tar.gz: bdfb147517579c653bed584d6437766b9d8fb26361f06d1688fa6d6d70c2f47709ed935d723c6cf9ccb3ada9cdf5709efde5fcc44109cde9eade56594404d2bc

data/lib/chao92/estimator.rb

@@ -2,9 +2,47 @@ module Chao92
 
   class Estimator
     
+    def initialize
+      @samples = []
+      @observed = {}
+      @ff = {}
+    end
+
     def self.run(samples)
-      1
-    end  
+      Estimator.new.sampling(samples)
+    end
+
+    def sampling(samples)
+      @samples += samples
+      update()
+      estimate()
+      @N
+    end
+
+    private
+    def update
+      @observed = @samples.to_hash
+      @ff = @observed.to_inverse
+    end
+
+    def estimate
+      @turing_estimator = if @ff[1] 
+                            1 - @ff[1].to_f / @samples.size 
+                          else         
+                            1      
+                          end
+
+      @N1 = @observed.size / @turing_estimator
+
+      tmp = 0.0
+      tmp = @samples.size.times do |x| 
+        tmp += x * (x - 1) * @ff[x] if @ff[x]
+      end
+
+      @cv_estimator = [@N1 * tmp / (@samples.size.to_f * (@samples.size - 1).to_f - 1), 0.0].max
+
+      @N = @N1 + @samples.size * (1 - @turing_estimator) / @turing_estimator * @cv_estimator
+    end
 
   end
 

data/lib/chao92/version.rb

@@ -1,3 +1,3 @@
 module Chao92
-  VERSION = "0.0.2"
+  VERSION = "1.0.0"
 end

data/lib/chao92.rb

@@ -1,3 +1,5 @@
+require 'support/core_ext'
+
 require "chao92/version"
 require "chao92/estimator"
 

data/lib/support/core_ext.rb

@@ -0,0 +1,21 @@
+class Array   
+  def to_hash     
+    hash = {}     
+    self.each {|e| hash[e] = (hash[e]? hash[e] + 1 : 1) }     
+    hash   
+  end 
+end
+
+class Hash   
+  def to_inverse     
+    ff = {}     
+    self.each do |k, v|        
+      if ff[v]
+        ff[v] += 1       
+      else         
+        ff[v] = 1       
+      end     
+    end     
+    ff
+  end 
+end

data/test/estimator_test.rb

@@ -1,13 +1,18 @@
 require 'test_helper'
 
-describe Chao92::Estimator do
+class TestEstimator < Minitest::Test
 
-  before do
+  def setup
+    @estimator = Chao92::Estimator.new
     @samples = [1, 1, 3, 5, 3, 3, 2, 1, 2, 3]
   end
 
-  it "test_estimator" do
-    assert_equal Chao92::Estimator.run(@samples), 1 
+  def test_estimator_run
+    assert_equal Chao92::Estimator.run(@samples).class, Float 
+  end
+
+  def test_estimator_sampling
+    assert_equal @estimator.sampling(@samples), Chao92::Estimator.run(@samples)
   end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chao92
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 1.0.0
 platform: ruby
 authors:
 - Eval Air
@@ -68,6 +68,7 @@ files:
 - lib/chao92.rb
 - lib/chao92/estimator.rb
 - lib/chao92/version.rb
+- lib/support/core_ext.rb
 - test/estimator_test.rb
 - test/test_helper.rb
 homepage: ''