blob: f3ffcd2ebf3e336879856e9403ca8820e7477e7b (
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
|
#!/usr/bin/env ruby
require 'zipf'
def dot v, w
sum = 0.0
v.each_with_index { |k,i|
sum += k * w[i]
}
return sum
end
def elen v
len = 0.0
v.each { |i| len += i**2 }
return Math.sqrt len
end
def norm v
len = elen v
return v.map { |i| i/len }
end
STDERR.write "loading feature dict\n"
fd = Marshal.load ReadFile.read ARGV[0]
d = fd.size
STDERR.write "#{d}\n"
STDERR.write "loading model\n"
w = Marshal.load ReadFile.read ARGV[1]
STDERR.write "predicting..\n"
err = 0
loss = 0.0
i = 0
while line = STDIN.gets
x = [0.0] * d
line.split.each { |i|
k,v = i.split '=', 2
x[fd[k]] = v.to_f
}
m = dot(w, norm(x))
if m <= 0.0
puts -1
loss += m.abs
err += 1
else
puts 1
end
i += 1
end
STDERR.write "#{err}/#{test.size}% accuracy, loss=#{loss}\n"
|