diff options
| author | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-11 12:13:40 +0200 |
|---|---|---|
| committer | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-11 12:13:40 +0200 |
| commit | de763211bd2b6451e3a8dc20eb95a3ecf72bef17 (patch) | |
| tree | e33da702f80d70dfa424cecea0a6495bab0fcfc2 /lib/classification.rb | |
| parent | 8b31acab67e22f30a87c995a94f1ee1e2a3d510f (diff) | |
initial dataset batch prediction
Diffstat (limited to 'lib/classification.rb')
| -rw-r--r-- | lib/classification.rb | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/classification.rb b/lib/classification.rb index 2668e4a..468c06a 100644 --- a/lib/classification.rb +++ b/lib/classification.rb @@ -20,10 +20,10 @@ module OpenTox end # DG: hack to ensure always two probability values # TODO: does not work for arbitrary feature names FIX!! - 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 +# 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) |