我们可以将固定的一组负样本传递给张量流损失或采样的softmax损失吗?

时间:2019-07-16 02:20:39

标签: python tensorflow word2vec word-embedding

我正在尝试使用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参数中的样本列表,而不是通过采样器。

0 个答案:

没有答案