diff options
author | Christoph Helma <helma@in-silico.ch> | 2016-10-06 19:14:10 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2016-10-06 19:14:10 +0200 |
commit | 4348eec89033e6677c9f628646fc67bd03c73fe6 (patch) | |
tree | 5b7465c3c286ba343268ce8e29b10cc1b03a667b /lib/train-test-validation.rb | |
parent | ec87f7e079f3a7ef8ea6a0fa57f3b40e81ecaed0 (diff) |
nano caret regression fixed
Diffstat (limited to 'lib/train-test-validation.rb')
-rw-r--r-- | lib/train-test-validation.rb | 5 |
1 files changed, 1 insertions, 4 deletions
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 |