tensorflow/word2vec运行报错tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be..
应该是tensorflow的版本不同,API的参数顺序有变动,源代码里的部分需要修改,改动代码如下:
loss = tf.reduce_mean(
tf.nn.nce_loss(weights=nce_weights,
biases=nce_biases,
inputs=embed,
labels=train_labels,
n
原创
2021-03-31 20:47:51 ·
637 阅读 ·
0 评论