ROC plot updated for readability

3 files changed, 12 insertions, 10 deletions

diff --git a/scripts/results2csv.rb b/scripts/results2csv.rb
deleted file mode 100755
index cb4d550..0000000
--- a/scripts/results2csv.rb
+++ /dev/null
@@ -1,9 +0,0 @@
-#!/usr/bin/env ruby
-require "yaml"
-
-data = YAML.load(File.read ARGV[0])
-puts "tpr,fpr"
-data.each do |algo,values|
-  algo = algo.sub("tensorflow","TF").sub("selected","FS").sub("lazar","L").sub("padel","P").sub("high-confidence","HC").sub("-all","").sub(".v3","")
-  puts [algo,values[:tpr],values[:fpr]].join(",")
-end
diff --git a/scripts/roc.R b/scripts/roc.R
index afc8293..459252a 100755
--- a/scripts/roc.R
+++ b/scripts/roc.R
@@ -2,7 +2,9 @@
 library(ggplot2)
 data <- read.csv("figures/roc.csv",header=T)
 p <- ggplot(data, aes(x=fpr, y=tpr)) + geom_abline()
-p <- p + geom_label(label=rownames(data) )
+#p <- p + geom_label(label=rownames(data) )
+p <- p + geom_point(aes(color=rownames(data)))
+p <- p + theme(legend.title=element_blank())
 p <- p + expand_limits(x=c(0,1),y=c(0,1))
 p <- p + labs(x = "False positive rate", y = "True positive rate")
 ggsave("figures/roc.png")
diff --git a/scripts/summary2roc.rb b/scripts/summary2roc.rb
new file mode 100755
index 0000000..dbac2f4
--- /dev/null
+++ b/scripts/summary2roc.rb
@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+require "yaml"
+
+data = YAML.load(File.read ARGV[0])
+puts "tpr,fpr"
+data.each do |algo,values|
+  algo = algo.sub("tensorflow","Tensorflow").sub("selected","FS").sub(".v3","").sub("-all"," (all)").sub("-high-confidence"," (high confidence)").sub("padel","PaDEL").sub("lazar ","lazar-MP2D ").sub("lr2","LR (scikit)").sub("lr","LR (SGD)").sub("nn","NN").sub("-rf","-RF")
+  puts [algo,values[:tpr],values[:fpr]].join(",")
+end