在用N-gram模型和词嵌入做单词预测的时候,经过网络预测之后net(input)之后,找出最大概率单词的坐标,pred_label_idx = out.max(1)[1].data[0] predict_word = idx_to_word[pred_label_idx] 并将网络预测的单词打印出来 出错:TypeError: unhashable type: 'numpy.ndarray' 出错原因:predict_word = idx_to_word[pred_label_idx] 读取字典值的时候出错,代码修改:pred_label_idx = out.max(1)[1].data[0].tolist() 这样错误就消失了。
网友评论