azami 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7ab05804a9638e3b584242d1ceade1e1595a7cb9
-  data.tar.gz: 3f67fcfb9d870e8b4c6b6a765f7c1c6843bb6842
+  metadata.gz: aff3ce034f65417a0d6a486e0e56e5bbe2ab20e6
+  data.tar.gz: 6c85bd9ffb21f9cc3f1309c1d8df6bd2e2e12849
 SHA512:
-  metadata.gz: b7fca74c5f0ead536db8bfd462037e204f20ad4d8c3f6d418b07d266e33b26e77925e97938a232a527f9193548e3e391196d75b4e629f47b8d9139341088adfe
-  data.tar.gz: 45d81c1dfa3fdc229dd001f730199981a04946f5a8d76f0396aa7b7fc333d6ce60b824d0063c36a1cb5815ea6e2e0a2aaf80d4ad96496d096941451136c74bc9
+  metadata.gz: 1c4b6634739be5c8ee532c26c92692d9811bc39c119794e51de011d0e79a5a95bf2cd405d9441c4e8456ffc21d3b4265b899715dacfff3fe839973f7f21d1f60
+  data.tar.gz: ce07429d25995a3f763039b66fe64177c4421ff6dbc5de22a1af6e56bd938c84c720eb0ce8e311fe651956f309eea0fcdb6a522b0a2e4fc5cbb0a934aa67541a

data/lib/azami.rb

@@ -1,32 +1,35 @@
+#Azami
 module Azami
-
   def self.em(data)
+    inputs = data[:inputs]
+    normal_params = data[:normal_params]
+    iteration_count = data[:iteration_count]
 
-    inputs, normal_params, iteration_count =
-      data[:inputs], data[:normal_params], data[:iteration_count]
-
-    iteration_count.times{
+    iteration_count.times do
       likelihood_weights =
-        inputs.map { |input|
-          likelihoods = normal_params.map { |normal_param|
-            mean,stdev = normal_param[:mean],normal_param[:stdev]
-            (0.398942*(2.71828**(-(0.5*(input - mean)**2)/stdev**2)))/stdev
-          }
+        inputs.map do |input|
+          likelihoods = normal_params.map do |normal_param|
+            mean = normal_param[:mean]
+            stdev = normal_param[:stdev]
+            (0.398942 * (2.71828**(-(0.5 * (input - mean)**2) / stdev**2))) / stdev
+          end
           total_likelihood = likelihoods.sum
-          likelihoods.map { |lk|
-            lk/total_likelihood
-          }
-        }.transpose
+          likelihoods.map do |lk|
+            lk / total_likelihood
+          end
+        end.transpose
 
-      normal_params.each_with_index { |normal_param, normal_index|
+      normal_params.each_with_index do |normal_param, normal_index|
         weights = likelihood_weights[normal_index]
         weights_sum = weights.sum
-        normal_param[:mean] = mean = (weights.map.with_index { |weight,input_i|
-          (weight*inputs[input_i])}.sum)/(weights_sum)
-        normal_param[:stdev] = Math.sqrt(((weights.map.with_index { |weight,input_i|
-          (((inputs[input_i]-mean)**2)*weight) }).sum)/(weights_sum))
-      }
-    }
+        normal_param[:mean] = mean = weights.map.with_index do |weight, input_i|
+                                        (weight * inputs[input_i])
+                                      end .sum / weights_sum
+        normal_param[:stdev] = Math.sqrt((weights.map.with_index do |weight, input_i|
+                                             (((inputs[input_i] - mean)**2) * weight)
+                                           end).sum / weights_sum)
+      end
+    end
     normal_params
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: azami
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - forvelin