summaryrefslogtreecommitdiff
path: root/scripts/roc.rb
blob: 8c8c93cdb0c0de716bdcf9d0a2e45e1fbb26fe1f (plain)
1
2
3
4
5
6
7
8
9
#!/usr/bin/env ruby
require "yaml"

data = YAML.load(File.read ARGV[0])[:cv]
puts "descriptor,model,tpr,fpr"
data.each do |algo,values|
  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