Probability fix

Code printed log-probabilities before, but now prints correct ones.
macournoyer · Aug 17, 2016 · 9c6ab55 · 9c6ab55
1 parent 5e3f2a1
commit 9c6ab55
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eval.lua b/eval.lua
@@ -35,7 +35,7 @@ function printProbabilityTable(wordIds, probabilities, num)
     local line = "| "
     for i = 1, num do
       local word = dataset.id2word[wordId[i]]
-      line = line .. string.format("%-10s(%4d%%)", word, probabilities[p][i] * 100) .. "  |  "
+      line = line .. string.format("%-10s(%4d%%)", word, torch.exp(probabilities[p][i]) * 100) .. "  |  "
     end
     print(line)
   end
@@ -67,4 +67,4 @@ while true do
   io.write("you> ")
   io.flush()
   io.write(say(io.read()))
-end
+end