应该是tensorflow的版本不同,API的参数顺序有变动,源代码里的部分需要修改,改动代码如下:
loss = tf.reduce_mean(
tf.nn.nce_loss(weights=nce_weights,
biases=nce_biases,
inputs=embed,
labels=train_labels,
num_sampled=num_sampled,
num_classes=vocabulary_size))
另外,部分函数的位置发生了变动,例:
optimizer = tf.compat.v1.train.GradientDescentOptimizer(1.0).minimize(loss)