getting regression values from prediction datasets fixed

1 files changed, 2 insertions, 2 deletions

diff --git a/lib/ot_predictions.rb b/lib/ot_predictions.rb
index 641fd8f..63debc0 100644
--- a/lib/ot_predictions.rb
+++ b/lib/ot_predictions.rb
@@ -107,7 +107,7 @@ module Lib
               predicted_values << class_values.index(value)
               confidence_values << prediction_dataset.get_prediction_confidence(c, predicted_variable)
             else
-              value = prediction_dataset.get_value(c, predicted_variable)
+              value = prediction_dataset.get_predicted_regression(c, predicted_variable)
               begin
                 value = value.to_f unless value==nil or value.is_a?(Numeric)
               rescue
@@ -137,4 +137,4 @@ module Lib
     end
     
   end
-end
\ No newline at end of file
+end