diff options
| author | Christoph Helma <helma@in-silico.ch> | 2016-04-28 08:11:12 +0200 |
|---|---|---|
| committer | Christoph Helma <helma@in-silico.ch> | 2016-04-28 08:11:12 +0200 |
| commit | 32d767ee7cfcc19337892551906950621f348174 (patch) | |
| tree | 20b1f34e933f4f4fec053db6daeeaa99feebf993 /lib/validation.rb | |
| parent | cfc64a2966ab38698e499f0b44f41208ee77a07f (diff) | |
nanoparticle crossvalidation technically working
Diffstat (limited to 'lib/validation.rb')
| -rw-r--r-- | lib/validation.rb | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/lib/validation.rb b/lib/validation.rb index 6b515e4..68cb1a1 100644 --- a/lib/validation.rb +++ b/lib/validation.rb @@ -24,12 +24,12 @@ module OpenTox def self.create model, training_set, test_set, crossvalidation=nil - atts = model.attributes.dup # do not modify attributes from original model + 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.save - predictions = validation_model.predict test_set.compounds + predictions = validation_model.predict test_set.substances predictions.each{|cid,p| p.delete(:neighbors)} nr_unpredicted = 0 predictions.each do |cid,prediction| @@ -43,7 +43,7 @@ module OpenTox validation = self.new( :model_id => validation_model.id, :test_dataset_id => test_set.id, - :nr_instances => test_set.compounds.size, + :nr_instances => test_set.substances.size, :nr_unpredicted => nr_unpredicted, :predictions => predictions#.sort{|a,b| p a; b[3] <=> a[3]} # sort according to confidence ) |