From 4348eec89033e6677c9f628646fc67bd03c73fe6 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Thu, 6 Oct 2016 19:14:10 +0200 Subject: nano caret regression fixed --- lib/train-test-validation.rb | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'lib/train-test-validation.rb') diff --git a/lib/train-test-validation.rb b/lib/train-test-validation.rb index 286614a..e3f5905 100644 --- a/lib/train-test-validation.rb +++ b/lib/train-test-validation.rb @@ -9,10 +9,7 @@ module OpenTox def self.create model, training_set, test_set - atts = model.attributes.dup # do not modify attributes of the original model - atts["_id"] = BSON::ObjectId.new - atts[:training_dataset_id] = training_set.id - validation_model = model.class.create model.prediction_feature, training_set, atts + validation_model = model.class.create prediction_feature: model.prediction_feature, training_dataset: training_set, algorithms: model.algorithms validation_model.save predictions = validation_model.predict test_set.substances nr_unpredicted = 0 -- cgit v1.2.3