add valdiation attribute medianConfidence

2 files changed, 15 insertions, 1 deletions

diff --git a/lib/predictions.rb b/lib/predictions.rb
index 00a5cdf..eb13b31 100755
--- a/lib/predictions.rb
+++ b/lib/predictions.rb
@@ -1,6 +1,15 @@
 
 require "lib/prediction_data.rb"
 
+
+class Array
+  
+  def median #array has to be sorted!     
+    (self[size/2] + self[(size+1)/2]) / 2.0
+  end 
+  
+end
+
 module Lib
 
   module Util
@@ -658,6 +667,10 @@ module Lib
 
     # data for (roc-)plots ###################################################################################
     
+    def median_confidence
+      @confidence_values.median if confidence_values_available?
+    end
+    
      def get_roc_prediction_values(class_value)
       
       #puts "get_roc_values for class_value: "+class_value.to_s
diff --git a/lib/validation_db.rb b/lib/validation_db.rb
index 17bd506..29f2903 100755
--- a/lib/validation_db.rb
+++ b/lib/validation_db.rb
@@ -10,7 +10,7 @@ module Validation
                 :training_dataset_uri, :prediction_feature, :test_dataset_uri, :test_target_dataset_uri, 
                 :prediction_dataset_uri, :date ] 
   VAL_PROPS_SUM = [ :num_instances, :num_without_class, :num_unpredicted ]
-  VAL_PROPS_AVG = [:real_runtime, :percent_without_class, :percent_unpredicted ]
+  VAL_PROPS_AVG = [ :real_runtime, :percent_without_class, :percent_unpredicted, :median_confidence ]
   VAL_PROPS = VAL_PROPS_GENERAL + VAL_PROPS_SUM + VAL_PROPS_AVG
   
   # :crossvalidation_info
@@ -76,6 +76,7 @@ module Validation
     attribute :regression_statistics_yaml
     attribute :finished    
     attribute :prediction_data_yaml
+    attribute :median_confidence
     
     index :model_uri
     index :validation_type