summaryrefslogtreecommitdiff
path: root/scripts/summary2table.rb
blob: 267bb97a72368b9449b4e1e651b1f07d47da353f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/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")[:cv]

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 = ["rf","lr","lr2","nn"].collect{|n| "tensorflow-"+n+".v3"}
when "lazar"
  header = ["MP2D", "PaDEL"]
  mp2dkeys = ["lazar-all","lazar-high-confidence"]
  padelkeys = ["lazar-padel-all","lazar-padel-high-confidence"]
  puts ","+header.join(",")
  rows.each do |short,long|
    print long+","
    print mp2dkeys.collect{|k| data[k][short]}.join("/")+","
    puts padelkeys.collect{|k| data[k][short]}.join("/")
  end
  exit
end
puts ","+header.join(",")
rows.each do |short,long|
  print long+","
  puts keys.collect{|k| data[k][short]}.join(",")
end