1 files changed, 3 insertions, 2 deletions
diff --git a/scripts/roc.rb b/scripts/roc.rb
index 8e2dc51..8c8c93c 100755
--- a/scripts/roc.rb
+++ b/scripts/roc.rb
@@ -2,7 +2,8 @@
 require "yaml"
 
 data = YAML.load(File.read ARGV[0])[:cv]
-puts "tpr,fpr"
+puts "descriptor,model,tpr,fpr"
 data.each do |algo,values|
-  puts [algo.upcase.gsub('_','-').sub("HIGH-CONFIDENCE","HC"),values[:tpr],values[:fpr]].join(",")
+  desc,model = algo.split("_",2)
+  puts [desc.upcase,model.upcase.gsub('_','-').sub("HIGH-CONFIDENCE","HC").sub(/^LR$/,"LR-sgd").sub("LR2","LR-scikit"),values[:tpr],values[:fpr]].join(",")
 end