Skip to content

Commit b7f5362

Browse files
committed
added index code
1 parent 917342c commit b7f5362

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Bigrams.lua

+1
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ function Bigrams:updateOutput(input)
112112

113113
for i = 1, batchSize do
114114
self.batchdraw(self, self.output[i], self.J[input[i]], self.q[input[i]])
115+
self.output[i]:apply(function(x) return self.bigrams[input[i]]['index'][x] end)
115116
end
116117

117118
return self.output

0 commit comments

Comments
 (0)