blob: dad160776e30b2bf9821257ebefa22f64ab44e6a (
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
|
#!/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
|
