我正在尝试使用tensorflow tf.nn.nce_loss()损失函数来训练单词嵌入模型。记录的功能签名为
tf.nn.nce_loss(
weights,
biases,
labels,
inputs,
num_sampled,
num_classes,
num_true=1,
sampled_values=None,
remove_accidental_hits=False,
partition_strategy='mod',
name='nce_loss'
)
但是,在尝试自定义损失函数时,我想看看是否有任何方法可以输入自己的负样本集而不是使用某些采样函数,以便可以避免使用num_sampled
参数并使用sampled_values
参数中的样本列表,而不是通过采样器。