lstm_sentence_classifier icon indicating copy to clipboard operation
lstm_sentence_classifier copied to clipboard

Published 20 hours ago • verified publisher icon yuchenlin

Fix vocab.load_vectors call

Open gombru opened this issue 6 years ago • 1 comments

Due to a change in torchtext the call to load_vectors has to be updated.

gombru avatar Feb 15 '18 15:02 gombru

The index on predictions is no longer needed and crashes, since pred.data.max(1)[1].numpy() returns a list of floats.

gombru avatar Feb 16 '18 10:02 gombru