tiny_outcome 2.1.0 → 3.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ccac5bae17f87452576130e604477b25fbaaa70785e0a41894077e8896521a28
-  data.tar.gz: f108cdf6533898223d088a169645d2b0ddc897c857c725c713086f42519a9d74
+  metadata.gz: 7e887122fe969136656c57bf1e8a240df3500f0120694891be4f9ba26cf9c694
+  data.tar.gz: 92e30c4c3c72afc4942181ede0badc1865b297c0444a1c7ec7cce272e4a7dac8
 SHA512:
-  metadata.gz: 12b256504992f76ca4e6a22af2b5a9e24699062765f3bf1acb6f9d1e24f9fceefcbf865e856371ab4bf85c9a20fef6923e8582bb9c87b37571bbdba4d1271752
-  data.tar.gz: fedc09804f2e2c2007cf05e83f0971a2f54a5e9dcb18a2cf245fd5d907e60813f83de23e55024b323bd065cda7acc08d6504928ea1706871b465831326ac33fe
+  metadata.gz: 31bf45e93abd24309b440599f7006573dd6b2beb19a588234b6f15aaed4334bde062b252f8bdecf2a7f8b72fe3a42def7a55d38a225b8e37cb19af2dfc375977
+  data.tar.gz: b828c61ee05e08de9041bdb5b17a95005ef2cf315419bd3806dc9feec2efb5ba2138231bd8ff9d4bc74e5d02079848c4356d0dc779d02709a12623cccc6ce66d

data/lib/tiny_outcome.rb

@@ -30,7 +30,12 @@ class TinyOutcome
               :samples,
               :warmth,
               :warmup,
-              :value
+              :probability,
+              :one_count,
+              :value,
+              :min,
+              :max,
+              :avg
 
   WARM_FULL = :full
   WARM_TWO_THIRDS = :two_thirds
@@ -44,9 +49,15 @@ class TinyOutcome
   #   samples that we can trust the probability output
   def initialize(precision, warmup=WARM_FULL)
     @precision = precision
+    @probability = 0.0
+    @one_count = 0
     @samples = 0
+    @min = -1.0
+    @max = -1.0
+    @avg = -1.0
     @warmth = 0
-    @value = 0
+    @value = [0] * @precision
+    @value_index = 0
     @warmup = case warmup
               when WARM_FULL        then precision
               when WARM_TWO_THIRDS  then (precision / 3) * 2
@@ -59,35 +70,37 @@ class TinyOutcome
               end
   end
 
-  # add a sample to the historic outcomes. the new sample is added to the
-  # low-order bits. the new sample is literally left-shifted into the value. the
-  # only reason this is a custom method is because some metadata needs to be
-  # updated when a new sample is added
+  def numeric_value
+    (full? ? @value.rotate(@value_index) : @value)[..(samples-1)].join.to_i(2)
+  end
+
+  # add a sample to the historic outcomes
   def <<(sample)
     raise "Invalid sample: #{sample}" unless sample == 0 || sample == 1
 
-    @value = ((value << 1) | sample) & (2**precision - 1)
+    removing_one = full? && @value[(@value_index + 1) % @precision] == 1
+
+    @value[@value_index] = sample
+    @value_index = (@value_index + 1) % @precision
     @warmth += 1 unless warmth == warmup
-    @samples += 1 unless samples == precision
+    @samples += 1 unless full?
+
+    # percentage of 1s out of the existing samples
+    #
+    #               number of 1s
+    # probabilty = ---------------
+    #               total samples
+    @one_count -= 1 if removing_one
+    @one_count += 1 if sample == 1
+    @probability = @one_count / samples.to_f
 
-    value
+    @value
   end
 
   # true if #probability is >= percentage
   # false otherwise
   def winner_at?(percentage)
-    probability >= percentage
-  end
-
-  # float: 0.0-1.0
-  # percentage of 1s out of the existing samples
-  #
-  #               number of 1s
-  # probabilty = ---------------
-  #               total samples
-  def probability
-    return 0.0 if samples == 0
-    value.to_s(2).count('1') / samples.to_f
+    @probability >= percentage
   end
 
   # true if we've received at least warmup number of samples
@@ -107,6 +120,30 @@ class TinyOutcome
     samples == precision
   end
 
+  # updates, and memoizes, the min/max/avg numbers. if you read the min/max/avg
+  # attributes you are getting the MEMOIZED values.
+  def update_stats!
+    return if @samples == 0
+
+    @min = 1.0
+    @max = 0.0
+    @avg = 0.0
+
+    sum = 0.0
+    raw = (full? ? @value.rotate(@value_index) : @value)[..(samples-1)]
+    group_size = [@samples, 100].min
+    num_groups = @samples - group_size
+    raw.each_cons(group_size) do |samples_group|
+      curr = samples_group.count(1) / samples.to_f
+      sum += curr
+
+      @min = curr if curr < @min
+      @max = curr if curr > @max
+    end
+
+    @avg = sum / (@num_groups.to_f + 1.0)
+  end
+
   # convenient way to see what's up
   def to_hash
     [:value,

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: tiny_outcome
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 3.1.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-09-03 00:00:00.000000000 Z
+date: 2023-09-28 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: m
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement