summaryrefslogtreecommitdiff
path: root/scripts/summary2table.rb
blob: d0da0af22a8248563f1c269812f4eec09f8a177a (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
31
32
33
34
#!/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("crossvalidations/summary.yaml")[:cv]

case ARGV[0]
when "tensorflow"
  header = ["MP2D-RF","MP2D-LR-sgd","MP2D-LR-scikit","MP2D-NN","MP2D-SVM","CDK-RF","CDK-LR-sgd","CDK-LR-scikit","CDK-NN","CDK-SVM"]
  desc = ["mp2d","cdk"]
  algos = ["rf","lr","lr2","nn","svm"]
  keys = []
  desc.each do |d| 
    algos.each do |a|
      keys << "tensorflow-"+a+"-"+d
    end
  end
when "lazar"
  header = ["MP2D", "CDK"]
  mp2dkeys = ["lazar-mp2d-all","lazar-mp2d-high-confidence"]
  cdkkeys = ["lazar-cdk-all","lazar-cdk-high-confidence"]
  puts ","+header.join(",")
  rows.each do |short,long|
    print long+","
    print mp2dkeys.collect{|k| data[k][short]}.join("/")+","
    puts cdkkeys.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