summaryrefslogtreecommitdiff
path: root/lib/train-test-validation.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/train-test-validation.rb')
-rw-r--r--lib/train-test-validation.rb2
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/train-test-validation.rb b/lib/train-test-validation.rb
index 8231728..bffee8c 100644
--- a/lib/train-test-validation.rb
+++ b/lib/train-test-validation.rb
@@ -28,7 +28,7 @@ module OpenTox
end
predictions.select!{|cid,p| p[:value] and p[:measurements]}
# hack to avoid mongos file size limit error on large datasets
- predictions.each{|cid,p| p[:neighbors] = []} #if model.training_dataset.name.match(/mutagenicity/i)
+ predictions.each{|cid,p| p.delete(:neighbors)} #if model.training_dataset.name.match(/mutagenicity/i)
validation = self.new(
:model_id => validation_model.id,
:test_dataset_id => test_set.id,