From d9c9d78e49d886ea91386adbbd2b523347df226e Mon Sep 17 00:00:00 2001
From: "helma@in-silico.ch" <helma@in-silico.ch>
Date: Mon, 29 Oct 2018 20:34:39 +0100
Subject: dataset predictions fixed

---
 lib/feature.rb | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'lib/feature.rb')

diff --git a/lib/feature.rb b/lib/feature.rb
index be07e7a..c18b0b8 100644
--- a/lib/feature.rb
+++ b/lib/feature.rb
@@ -54,18 +54,30 @@ module OpenTox
   class NominalLazarPrediction < NominalFeature
     field :model_id, type: BSON::ObjectId
     field :training_feature_id, type: BSON::ObjectId
+    def name
+      "#{self[:name]} Prediction"
+    end
   end
 
   class LazarPredictionProbability < NominalLazarPrediction
+    def name
+      "probability(#{self[:name]})"
+    end
   end
 
   # Numeric lazar prediction
   class NumericLazarPrediction < NumericFeature
     field :model_id, type: BSON::ObjectId
     field :training_feature_id, type: BSON::ObjectId
+    def name
+      "#{name} Prediction"
+    end
   end
 
   class LazarPredictionInterval < NumericLazarPrediction
+    def name
+      "prediction_interval_#{self[:name]}"
+    end
   end
 
   class NominalSubstanceProperty < NominalFeature
-- 
cgit v1.2.3