search

qiaoguan / deep-ctr-prediction

CTR prediction models based on deep learning(基于深度学习的广告推荐CTR预估模型)
https://github.com/qiaoguan/deep-ctr-prediction
907 stars 276 forks source link

预测代码无法运行 #8

Closed 378994621 closed 4 years ago

378994621 commented 4 years ago

user_id = features['user_id'] click_label = features['label'] conversion_label = features['is_conversion']

if mode == tf.estimator.ModeKeys.PREDICT: predictions = { 'ctr_preds': ctr_preds, 'cvr_preds': cvr_preds, 'ctcvr_preds': ctcvr_preds, 'user_id': user_id, 'click_label': click_label, 'conversion_label': conversion_label } 以上代码为ESSM第37-50行，从代码逻辑来看click_label，conversion_label是预测的对象，应该对应网络中的tensor啊，为什么从features直接读取啊？我的理解这两个label应该是分别从ctr_preds，ctcvr_preds转换过来的吧

qiaoguan commented 4 years ago

label是训练时候用的， ctr_preds, ctcvr_preds 这个是模型训练好之后预测用的