diff options
Diffstat (limited to 'lib/classification.rb')
-rw-r--r-- | lib/classification.rb | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/classification.rb b/lib/classification.rb index 0a32126..b4b2e59 100644 --- a/lib/classification.rb +++ b/lib/classification.rb @@ -11,7 +11,7 @@ module OpenTox confidence = 0.0 neighbors.each do |row| n,sim,acts = row - confidence = sim if sim > confidence # distance to nearest neighbor + #confidence = sim if sim > confidence # distance to nearest neighbor acts.each do |act| weighted_sum[act] ||= 0 weighted_sum[act] += sim @@ -24,7 +24,7 @@ module OpenTox sim_sum = weighted_sum[weighted_sum.keys[0]] sim_sum -= weighted_sum[weighted_sum.keys[1]] sim_sum > 0 ? prediction = weighted_sum.keys[0] : prediction = weighted_sum.keys[1] - #confidence = (sim_sum/neighbors.size).abs + confidence = (sim_sum/neighbors.size).abs return {:value => prediction,:confidence => confidence} else bad_request_error "Cannot predict more than 2 classes, multinomial classifications is not yet implemented. Received classes were: '#{weighted.sum.keys}'" |