#!/usr/bin/env ruby
require 'yaml'

rows = {:acc => "Accuracy", :tpr => "True positive rate/Sensitivity", :tnr => "True negative rate/Specificity", :ppv => "Positive predictive value/Precision", :npv => "Negative predictive value", :n => "Nr. predictions"}
data = YAML.load_file "10-fold-crossvalidations/summary.yaml"

case ARGV[0]
when "R"
  header = ["RF","SVM","DL"]
  keys = header.collect{|h| "R-"+h}
when "tensorflow"
  header = ["RF","LR (SGD)","LR (SCIKIT)","NN"]
  keys = ["lr","lr2","nn"].collect{|n| "tensorflow-"+n+".v3"}
when "lazar"
  header = ["lazar-mp2d (all)","lazar-mp2d (high confidence)", "lazar-padel (all)","lazar-padel (high confidence)"]
  keys = ["lazar-all","lazar-high-confidence", "lazar-padel-all","lazar-padel-high-confidence"]
end
puts ","+header.join(",")
rows.each do |short,long|
  print long+","
  puts keys.collect{|k| data[k][short]}.join(",")
end
exit