diff options
author | Christoph Helma <helma@in-silico.ch> | 2016-12-20 16:28:36 +0100 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2016-12-20 16:28:36 +0100 |
commit | a5abdd27b8c9b3f1cf65a567bfd9eb17ecc91a72 (patch) | |
tree | 486c73a7e498f82ef49b140ffa6f2c94e63c7b4e /lib/validation-statistics.rb | |
parent | 68a87da10774a1af117d075b26302b0f8b85bf19 (diff) |
true_rate, predictivity stored in crossvalidationnano-lazar-paper.submission
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 |