diff options
Diffstat (limited to 'lib/classification.rb')
| -rw-r--r-- | lib/classification.rb | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lib/classification.rb b/lib/classification.rb index 93b4f0f..4cc9201 100644 --- a/lib/classification.rb +++ b/lib/classification.rb @@ -6,13 +6,14 @@ module OpenTox def self.weighted_majority_vote compound, params neighbors = params[:neighbors] feature_id = params[:prediction_feature_id].to_s + dataset_id = params[:training_dataset_id].to_s sims = {} neighbors.each do |n| sim = n["tanimoto"] - n["toxicities"][feature_id].each do |act| + n["toxicities"][feature_id][dataset_id].each do |act| sims[act] ||= [] sims[act] << sim - end + end if n["toxicities"][feature_id][dataset_id] end sim_all = sims.collect{|a,s| s}.flatten sim_sum = sim_all.sum |