summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authormguetlein <martin.guetlein@gmail.com>2011-07-06 14:57:53 +0200
committermguetlein <martin.guetlein@gmail.com>2011-07-06 14:57:53 +0200
commit9407fcd5c937370324764a08afd589f386e83b60 (patch)
tree516a274a692e55dbbe5bfa9f439ccc536627b452
parentd51a6d01d2a54512f1969e3ef98f5beb24c643e9 (diff)
fix error when making regression validations w/o confidence
-rwxr-xr-xlib/predictions.rb4
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/predictions.rb b/lib/predictions.rb
index 4e34c90..082ad10 100755
--- a/lib/predictions.rb
+++ b/lib/predictions.rb
@@ -146,9 +146,9 @@ module Lib
delta = predicted_value - actual_value
@sum_error += delta
@sum_abs_error += delta.abs
- @sum_weighted_abs_error += delta.abs*confidence_value
+ @sum_weighted_abs_error += delta.abs*confidence_value if @conf_provided
@sum_squared_error += delta**2
- @sum_weighted_squared_error += (delta**2)*confidence_value
+ @sum_weighted_squared_error += (delta**2)*confidence_value if @conf_provided
old_prediction_mean = @prediction_mean
@prediction_mean = (@prediction_mean * (@num_predicted-1) + predicted_value) / @num_predicted.to_f