summaryrefslogtreecommitdiff
path: root/lib/model.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/model.rb')
-rw-r--r--lib/model.rb10
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/model.rb b/lib/model.rb
index 4aa8642..0155fc8 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -16,6 +16,10 @@ module OpenTox
field :prediction_algorithm, type: String
# prediction feature
field :prediction_feature_id, type: BSON::ObjectId
+
+ def training_dataset
+ Dataset.find(training_dataset_id)
+ end
end
class Lazar < Model
@@ -24,7 +28,7 @@ module OpenTox
field :neighbor_algorithm, type: String
field :neighbor_algorithm_parameters, type: Hash
- attr_accessor :prediction_dataset
+ #attr_accessor :prediction_dataset
#attr_accessor :training_dataset
# Create a lazar model from a training_dataset and a feature_dataset
@@ -115,10 +119,6 @@ module OpenTox
return prediction_dataset
end
- def training_dataset
- Dataset.find(training_dataset_id)
- end
-
end
def training_activities