From 34d621606e58adeb35aa33f8503e105189a7c0d5 Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Fri, 8 Apr 2016 23:19:28 +0200
Subject: latest stuff

---
 rerank_output_lists.rb | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100755 rerank_output_lists.rb

(limited to 'rerank_output_lists.rb')

diff --git a/rerank_output_lists.rb b/rerank_output_lists.rb
new file mode 100755
index 0000000..22bd9ca
--- /dev/null
+++ b/rerank_output_lists.rb
@@ -0,0 +1,48 @@
+#!/usr/bin/env ruby
+
+require 'zipf'
+
+class KbestItem
+  attr_accessor :rank, :model, :rr, :gold, :f
+  def initialize s
+    a = s.split "\t"
+    @rank = a[0].to_i
+    @gold = a[1].to_f
+    @model = a[2].to_f
+    @rr    = -1.0
+    @f = SparseVector.from_kv a[3], "=", " "
+  end
+
+  def to_s
+    return "#{@model}\t#{@gold}\t#{@rank}\t#{@rr}"
+  end
+end
+
+w = SparseVector.from_kv ReadFile.new(ARGV[0]).read, "\t", "\n"
+
+def o kl, j
+  f = WriteFile.new "lists/#{j}.model"
+  g = WriteFile.new "lists/#{j}.rr"
+  kl.each { |i|
+    f.write "#{i.gold}\t#{i.model}\n"
+    g.write "#{i.gold}\t#{i.rr}\n"
+  }
+end
+
+`mkdir -p lists`
+STDERR.write "reranking..\n"
+cur = []
+k_sum = 0
+j = 0
+while line = STDIN.gets
+  item = KbestItem.new line.strip
+  item.rr = w.dot(item.f)
+  if item.rank == 0 && cur.size > 0
+    o cur, j
+    cur = []
+    j += 1
+  end
+  cur << item
+end
+o cur, j
+
-- 
cgit v1.2.3