RandomForest - replace sum with reduce +

asafschers · Oct 13, 2017 · 54d82f2 · 54d82f2
1 parent 5219290
commit 54d82f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/scoruby/models/random_forest.rb b/lib/scoruby/models/random_forest.rb
@@ -22,7 +22,7 @@ def predict(features)
         decisions_count = decisions_count(features)
         {
             label: decisions_count.max_by {|_, v| v}[0],
-            score: decisions_count.max_by {|_, v| v}[1] / decisions_count.values.sum.to_f
+            score: decisions_count.max_by {|_, v| v}[1] / decisions_count.values.reduce(0, :+).to_f
         }
       end
     end