修改损失函数

tensnsorflow/train.py

@@ -180,7 +180,8 @@ def model_fn(features, labels, mode, params):
         ctr_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=y_ctr, labels=y))
         #cvr_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=y_ctcvr, labels=z))
         cvr_loss = tf.reduce_mean(tf.losses.log_loss(predictions=pctcvr, labels=z))
-        loss = ctr_task_wgt * ctr_loss + (1  -ctr_task_wgt) * cvr_loss + l2_reg * tf.nn.l2_loss(Feat_Emb)
+        # loss = ctr_task_wgt * ctr_loss + (1  -ctr_task_wgt) * cvr_loss + l2_reg * tf.nn.l2_loss(Feat_Emb)
+        loss = ctr_loss
 
         tf.summary.scalar('ctr_loss', ctr_loss)
         tf.summary.scalar('cvr_loss', cvr_loss)