blob: 684825dff965387c87c4eb6adb7124345443bc1d (
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#!/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}"
end
end
ws = []
cs = []
ReadFile.readlines_strip(ARGV[0]).each { |l|
c, s = l.split "\t"
cs << c.to_i
next if !s||s.strip==""
ws << SparseVector.from_kv(s, "=", " ")
}
def sign(x)
if x <= 0
return -1.0
else
return 1.0
end
end
def o kl
scores = []
scores << kl.first.gold
kl.sort! { |i,j| j.model <=> i.model }
scores << kl.first.gold
kl.sort! { |i,j| j.rr <=> i.rr }
scores << kl.first.gold
puts scores.join "\t"
end
STDERR.write "reranking..\n"
cur = []
k_sum = 0
j = 0
while line = STDIN.gets
item = KbestItem.new line.strip
item.rr = 0
ws.each_with_index{ |w,j|
item.rr += sign(w.dot(x))*cs[j]
}
if item.rank == 0 && cur.size > 0
o cur
cur = []
j += 1
end
cur << item
end
o cur
|
