在Tensorflow中,我正在用完全连接的多层神经网络进行多元回归。该网络在给定输入向量(y1,y2)的情况下,预测2个连续浮动变量(x1,x2,...xN),即网络有2个输出节点。对于两个输出,网络似乎没有收敛。我的损失函数实质上是预测向量和真理向量(每个包含两个标量)之间的L2距离:
loss = tf.nn.l2_loss(tf.sub(
我想在BERT中添加一个输出层,以预测多个二进制特性。是肯定的吗?1如果是,0否则。是关于体育的吗?1如果是,0如果不是。是关于商业的吗?1如果是,0如果不是。我的第一个想法是把它作为一个回归任务,增加一个输出层,包含3个神经元:每个问题一个。(nn.Linear(dim_in, 3))
def forward(self, in