tf.nn.sigmoid_cross_entropy_with_logits它是否共享权重?

时间:2018-02-01 14:00:08

标签: tensorflow neural-network classification multilabel-classification

我打算使用

tf.nn.sigmoid_cross_entropy_with_logits 

用于创建N个二进制分类模型。我希望这些N模型是独立的二元模型而不是共享权重?我可以使用此功能实现吗?

2 个答案:

答案 0 :(得分:1)

是的,你可以,这个函数只是将sigmoid应用于给定的logits,然后计算交叉熵损失。它根本没有重量。

答案 1 :(得分:0)

如果我理解你的问题,如果你有几个模型,你需要为每个模型分别设置一个损失函数,以便分别优化每个模型。在这种情况下,您应该为每个模型分别使用sigmoid_cross_entropy_with_logits。每个模型都会在此函数中放置自己的logits,然后使用优化器将其最小化。