diff options
author | Christoph Helma <helma@in-silico.ch> | 2016-03-15 18:46:34 +0100 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2016-03-15 18:46:34 +0100 |
commit | abc3526e318a2bfa24dfe033d8879e7657c2ae5c (patch) | |
tree | b11500952085d6a771f845431258c7befae5fd04 /lib/model.rb | |
parent | 7c3bd90c26dfeea2db3cf74a1cefc23d8dece7c0 (diff) |
single tests pass
Diffstat (limited to 'lib/model.rb')
-rw-r--r-- | lib/model.rb | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/model.rb b/lib/model.rb index f21ea54..5da5dc8 100644 --- a/lib/model.rb +++ b/lib/model.rb @@ -66,6 +66,7 @@ module OpenTox prediction.merge!({:value => nil,:confidence => nil,:warning => "Could not find similar compounds with experimental data in the training dataset."}) else prediction.merge!(Algorithm.run(prediction_algorithm, compound, {:neighbors => neighbors,:training_dataset_id=> training_dataset_id,:prediction_feature_id => prediction_feature.id})) + prediction[:neighbors] = neighbors end prediction end @@ -95,7 +96,7 @@ module OpenTox case object.class.to_s when "OpenTox::Compound" prediction = predictions.first - prediction[:neighbors] = neighbors.sort{|a,b| b[1] <=> a[1]} # sort according to similarity + prediction[:neighbors].sort!{|a,b| b[1] <=> a[1]} # sort according to similarity return prediction when "Array" return predictions |