bayesnet 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9f17178cdfa472aea14e7769f1831f1903ceee2613b6e7b71e1135990dc974ab
-  data.tar.gz: c55cb95134d4dbb479e89839be7a4634b6bc6a8be52c5c066f1cc1ad21c584a7
+  metadata.gz: ca0c7b763a97d4516fb893a7c79716bd4f7279be3fba1b0a9d55f43279aa95cf
+  data.tar.gz: 55da869a7d0436bd7eb08a8787bc5d19d56e71e90ac283f4568931c391d3f407
 SHA512:
-  metadata.gz: fda426d9dd319b2ad8300a8ba59acae511740cc271729f40b36ce7a7765d8019a901d9d83a2e955ccd964901ee604a3ad96fba092ce41394da5980a77d1bb198
-  data.tar.gz: 03d4beb66b11ba684007dd49f67ef2ae085ecd5c6e5ca9b8b63327eee109ada3b1f389731543fa6d970bff110230231bc1533022640367a906f57dec1c41bc6d
+  metadata.gz: 203475b19fc5bfa4151b4586243ebedce938fcf6a5c5ed3d561f7553ea011e3c8cc797d45360be790f691cecf7462623a0e179160d770454a5bc5cd8ed1a77a2
+  data.tar.gz: fd014b35e9f98908694b5a25655f279b8885f01e59e30edf17c34a433758dca7d5c271b6318b5d67df921ce39a63c0e643fb84de241064d8bf8ec95e9a39b94d

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    bayesnet (0.0.2)
+    bayesnet (0.0.3)
 
 GEM
   remote: https://rubygems.org/

data/lib/bayesnet/factor.rb

@@ -8,25 +8,27 @@ module Bayesnet
     end
 
     # Specifies variable name together with its values
-    def var(var_name_to_values)
-      @vars.merge!(var_name_to_values)
+    def scope(var_name_to_values)
+      @scope.merge!(var_name_to_values)
     end
 
-    # Specifies function values for args. Latest args is an function value, all previous are argument values
-    def val(*args)
-      args = args[0] if args.size == 1 && args[0].is_a?(Array)
-      @vals[args[0..-2]] = args[-1]
+    # Specifies value for a scope context. Value is the last element in `context_and_val`
+    def val(*context_and_val)
+      if context_and_val.size == 1 && context_and_val[0].is_a?(Array)
+        context_and_val = context_and_val[0]
+      end
+      @vals[context_and_val[0..-2]] = context_and_val[-1]
     end
 
     def var_names
-      @vars.keys
+      @scope.keys
     end
 
-    def [](*args)
-      key = if args.size == 1 && args[0].is_a?(Hash)
-        args[0].slice(*var_names).values
+    def [](*context)
+      key = if context.size == 1 && context[0].is_a?(Hash)
+        context[0].slice(*var_names).values
       else
-        args
+        context
       end
       @vals[key]
     end
@@ -35,9 +37,9 @@ module Bayesnet
       self.class.new(var_distribution.keys, var_distribution.values.map(&:to_a))
     end
 
-    def args(*var_names)
+    def contextes(*var_names)
       return [] if var_names.empty?
-      @vars[var_names[0]].product(*var_names[1..].map { |var_name| @vars[var_name] })
+      @scope[var_names[0]].product(*var_names[1..].map { |var_name| @scope[var_name] })
     end
 
     def values
@@ -48,42 +50,44 @@ module Bayesnet
       vals = @vals.clone
       norm_factor = vals.map(&:last).sum * 1.0
       vals.each { |k, v| vals[k] /= norm_factor }
-      self.class.new(@vars.clone, vals)
+      self.class.new(@scope.clone, vals)
     end
 
-    def limit_by(evidence)
+    def reduce_to(context)
       # todo: use Hash#except when Ruby 2.6 support no longer needed
-      evidence_keys_set = evidence.keys.to_set
-      vars = @vars.reject { |k, _| evidence_keys_set.include?(k) }
+      context_keys_set = context.keys.to_set
+      scope = @scope.reject { |k, _| context_keys_set.include?(k) }
 
-      evidence_vals = evidence.values
-      indices = evidence.keys.map { |k| index_by_var_name[k] }
-      vals = @vals.select { |k, v| indices.map { |i| k[i] } == evidence_vals }
-      vals.transform_keys! { |k| k - evidence_vals }
+      context_vals = context.values
+      indices = context.keys.map { |k| index_by_var_name[k] }
+      vals = @vals.select { |k, v| indices.map { |i| k[i] } == context_vals }
+      vals.transform_keys! { |k| k - context_vals }
 
-      self.class.new(vars, vals)
+      self.class.new(scope, vals)
     end
 
-    def reduce(over_vars)
-      vars = @vars.slice(*over_vars)
-      indices = vars.keys.map { |k| index_by_var_name[k] }
-      vals = @vals.group_by { |args, val| indices.map { |i| args[i] } }
+    # groups by `var_names` having same context and sum out values.
+    def marginalize(var_names)
+      scope = @scope.slice(*var_names)
+
+      indices = scope.keys.map { |k| index_by_var_name[k] }
+      vals = @vals.group_by { |context, val| indices.map { |i| context[i] } }
       vals.transform_values! { |v| v.map(&:last).sum }
-      reduced = self.class.new(vars, vals)
-      reduced.normalize
+
+      self.class.new(scope, vals)
     end
 
     private
 
-    def initialize(vars = {}, vals = {})
-      @vars = vars
+    def initialize(scope = {}, vals = {})
+      @scope = scope
       @vals = vals
     end
 
     def index_by_var_name
       return @index_by_var_name if @index_by_var_name
       @index_by_var_name = {}
-      @vars.each_with_index { |(k, v), i| @index_by_var_name[k] = i }
+      @scope.each_with_index { |(k, v), i| @index_by_var_name[k] = i }
       @index_by_var_name
     end
   end

data/lib/bayesnet/graph.rb

@@ -20,14 +20,16 @@ module Bayesnet
     end
 
     def distribution(over: [], evidence: {})
-      limited = joint_distribution.limit_by(evidence)
-      limited.reduce(over)
+      joint_distribution
+        .reduce_to(evidence)
+        .marginalize(over)
+        .normalize
     end
 
     # This is MAP query, i.e. Maximum a Posteriory
     def most_likely_value(var_name, evidence:)
       posterior_distribution = distribution(over: [var_name], evidence: evidence)
-      mode = posterior_distribution.args(var_name).zip(posterior_distribution.values).max_by(&:last)
+      mode = posterior_distribution.contextes(var_name).zip(posterior_distribution.values).max_by(&:last)
       mode.first.first
     end
 
@@ -47,15 +49,15 @@ module Bayesnet
 
       factor = Factor.new
       @nodes.each do |node_name, node|
-        factor.var node_name => node.values
+        factor.scope node_name => node.values
       end
 
-      factor.args(*var_names).each do |args|
-        val_by_name = var_names.zip(args).to_h
+      factor.contextes(*var_names).each do |context|
+        val_by_name = var_names.zip(context).to_h
         val = nodes.values.reduce(1.0) do |prob, node|
           prob * node.factor[val_by_name]
         end
-        factor.val args + [val]
+        factor.val context + [val]
       end
       @joint_distribution = factor.normalize
     end

data/lib/bayesnet/node.rb

@@ -16,7 +16,7 @@ module Bayesnet
         @values = hash_or_array.keys
         node = self
         @factor = Factor.build do
-          var node.name => node.values
+          scope node.name => node.values
           hash_or_array.each do |value, probability|
             val [value, probability]
           end
@@ -26,9 +26,9 @@ module Bayesnet
         @values = hash_or_array
         node = self
         @factor = Factor.build do
-          var node.name => node.values
+          scope node.name => node.values
           node.parent_nodes.each do |parent_node_name, parent_node|
-            var parent_node_name => parent_node.values
+            scope parent_node_name => parent_node.values
           end
         end
         instance_eval(&block)

data/lib/bayesnet/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Bayesnet
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bayesnet
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Aleksandr Furmanov
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-12-28 00:00:00.000000000 Z
+date: 2021-12-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: m