diff options
| author | gebele <gebele@in-silico.ch> | 2017-05-26 12:53:01 +0000 |
|---|---|---|
| committer | gebele <gebele@in-silico.ch> | 2017-05-26 12:53:01 +0000 |
| commit | 6ed197736516d98e200cc64d922f42eb3122589c (patch) | |
| tree | d2c7de76b020be254cc82563d36e6711fd6f1867 /lib/classification.rb | |
| parent | 61a7d994d8f4fbcf25414beea96189bf885ad19d (diff) | |
| parent | 9aa5203dd375225996c1efe4be1a4324ddc6cda7 (diff) | |
Merge branch 'development'
Diffstat (limited to 'lib/classification.rb')
| -rw-r--r-- | lib/classification.rb | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/lib/classification.rb b/lib/classification.rb index 638492b..a875903 100644 --- a/lib/classification.rb +++ b/lib/classification.rb @@ -18,6 +18,11 @@ module OpenTox class_weights.each do |a,w| probabilities[a] = w.sum/weights.sum end + # DG: hack to ensure always two probability values + if probabilities.keys.uniq.size == 1 + missing_key = probabilities.keys.uniq[0].match(/^non/) ? probabilities.keys.uniq[0].sub(/non-/,"") : "non-"+probabilities.keys.uniq[0] + probabilities[missing_key] = 0.0 + end probabilities = probabilities.collect{|a,p| [a,weights.max*p]}.to_h p_max = probabilities.collect{|a,p| p}.max prediction = probabilities.key(p_max) |