summaryrefslogtreecommitdiff
path: root/scripts/confusion-matrix-summary.rb
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/confusion-matrix-summary.rb')
-rwxr-xr-xscripts/confusion-matrix-summary.rb19
1 files changed, 19 insertions, 0 deletions
diff --git a/scripts/confusion-matrix-summary.rb b/scripts/confusion-matrix-summary.rb
new file mode 100755
index 0000000..a9742f7
--- /dev/null
+++ b/scripts/confusion-matrix-summary.rb
@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+require 'csv'
+require 'json'
+
+csv = CSV.read(ARGV[0],headers: false,:col_sep => ",")
+tp = csv[0][0].to_f
+fp = csv[0][1].to_f
+fn = csv[1][0].to_f
+tn = csv[1][1].to_f
+
+result = {
+ :accuracy => (tp+tn)/(tp+fp+tn+fn),
+ :true_positive_rate => tp/(tp+fn),
+ :true_negative_rate => tn/(tn+fp),
+ :positive_predictive_value => tp/(tp+fp),
+ :negative_predictive_value => tn/(tn+fn),
+}
+
+puts result.to_json