blob: 12a9f6f6ebc7a005ff31ff7ce4c659d4135955e0 (
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
|
#!/usr/bin/env ruby
require "zipf"
require "optimist"
def main
conf = Optimist::options do
opt :kbests, "kbests", :type => :string, :default => "-"
opt :references, "references", :type => :string, :required => true
end
refs = ReadFile.new conf[:references]
kbest_lists = read_kbest_lists conf[:kbests]
i = 0
kbest_lists.each { |list|
scores = []
o = false
list.each { |e| scores << per_sentence_bleu(e, refs[i]) }
max = scores.max
scores.each_with_index { |x,j|
puts "#{j+1} ||| #{scores[j]} ||| #{list[j]}"
if scores[j]==max && !o
puts "^^^ #{j+1} #{max}"
o = true
end
}
puts
i += 1
}
end
main
|
