1 files changed, 4 insertions, 0 deletions
diff --git a/lib/feature.rb b/lib/feature.rb
index e6fede6..2c10c26 100644
--- a/lib/feature.rb
+++ b/lib/feature.rb
@@ -37,6 +37,10 @@ module OpenTox
     field :training_feature_id, type: BSON::ObjectId
   end
 
+  class LazarPredictionProbability < NominalLazarPrediction
+    field :value, type: Float
+  end
+
   # Numeric lazar prediction
   class NumericLazarPrediction < NumericFeature
     field :model_id, type: BSON::ObjectId