1 files changed, 9 insertions, 0 deletions
diff --git a/lib/validation_db.rb b/lib/validation_db.rb
index be004fb..c3a3f71 100755
--- a/lib/validation_db.rb
+++ b/lib/validation_db.rb
@@ -72,6 +72,7 @@ module Validation
     attribute :classification_statistics_yaml
     attribute :regression_statistics_yaml
     attribute :finished    
+    attribute :prediction_data_yaml
     
     index :model_uri
     index :validation_type
@@ -99,6 +100,14 @@ module Validation
     def regression_statistics=(rs)
       self.regression_statistics_yaml = rs.to_yaml
     end
+    
+    def prediction_data
+      YAML.load(self.prediction_data_yaml) if self.prediction_data_yaml
+    end
+    
+    def prediction_data=(pd)
+      self.prediction_data_yaml = pd.to_yaml
+    end
 
     def save
       super