add additional files

1 files changed, 35 insertions, 0 deletions

diff --git a/prediction.rb b/prediction.rb
new file mode 100644
index 0000000..afceb08
--- /dev/null
+++ b/prediction.rb
@@ -0,0 +1,35 @@
+module OpenTox
+
+  class Prediction
+
+    include OpenTox
+    include Mongoid::Document
+    include Mongoid::Timestamps
+    store_in collection: "predictions"
+    field :compound, type: BSON::ObjectId
+    field :model, type: BSON::ObjectId
+    field :prediction, type: Hash, default:{}
+    field :csv, type: String
+
+    attr_accessor :compound, :model, :prediction, :csv
+
+    def compound
+      self[:compound]
+    end
+    
+    def model
+      self[:model]
+    end
+    
+    def prediction
+      self[:prediction]
+    end
+
+    def csv
+      self[:csv]
+    end
+
+  end
+
+end
+