diff options
author | gebele <gebele@in-silico.ch> | 2016-12-20 16:26:09 +0000 |
---|---|---|
committer | gebele <gebele@in-silico.ch> | 2016-12-20 16:26:09 +0000 |
commit | 76912e72cf52c2a10423226ababb2b6d7f4b84ca (patch) | |
tree | e167ec15c9b9434c027a341db50a3e7ea4955c34 /lib/validation-statistics.rb | |
parent | da086fad5b45c0d7b59feb40d0108ac620613933 (diff) | |
parent | a5abdd27b8c9b3f1cf65a567bfd9eb17ecc91a72 (diff) |
Merge branch 'development'1.0.0
Diffstat (limited to 'lib/validation-statistics.rb')
-rw-r--r-- | lib/validation-statistics.rb | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/lib/validation-statistics.rb b/lib/validation-statistics.rb index b6f8a60..2202b79 100644 --- a/lib/validation-statistics.rb +++ b/lib/validation-statistics.rb @@ -6,8 +6,6 @@ module OpenTox self.accept_values = model.prediction_feature.accept_values self.confusion_matrix = Array.new(accept_values.size){Array.new(accept_values.size,0)} self.weighted_confusion_matrix = Array.new(accept_values.size){Array.new(accept_values.size,0)} - true_rate = {} - predictivity = {} nr_instances = 0 predictions.each do |cid,pred| # TODO @@ -38,11 +36,11 @@ module OpenTox end end end - true_rate = {} - predictivity = {} + self.true_rate = {} + self.predictivity = {} accept_values.each_with_index do |v,i| - true_rate[v] = confusion_matrix[i][i]/confusion_matrix[i].reduce(:+).to_f - predictivity[v] = confusion_matrix[i][i]/confusion_matrix.collect{|n| n[i]}.reduce(:+).to_f + self.true_rate[v] = confusion_matrix[i][i]/confusion_matrix[i].reduce(:+).to_f + self.predictivity[v] = confusion_matrix[i][i]/confusion_matrix.collect{|n| n[i]}.reduce(:+).to_f end confidence_sum = 0 weighted_confusion_matrix.each do |r| @@ -60,8 +58,8 @@ module OpenTox :weighted_confusion_matrix => weighted_confusion_matrix, :accuracy => accuracy, :weighted_accuracy => weighted_accuracy, - :true_rate => true_rate, - :predictivity => predictivity, + :true_rate => self.true_rate, + :predictivity => self.predictivity, } end |