Skip to content

Latest commit

 

History

History

esmm

README.md

tf.version == '2.1.0'

Data format :

说明：

（1）第1列：ctr 的值，0表示曝光后未点击，1表示曝光后点击；
（2）第2列：ctcvr 的值，0表示（曝光后未点击、或曝光后点击）后未购买，1表示（曝光、点击）后购买；

（3）第3-130列：用户的预训练Embedding；
（4）第131-258列：商品的预训练Embedding；

run model

sh master.sh

模型 summary

__________________________________________________________________________________________________
Layer (type)                    Output Shape         Param #     Connected to
==================================================================================================
input_1 (InputLayer)            [(None, 128)]        0
__________________________________________________________________________________________________
input_2 (InputLayer)            [(None, 128)]        0
__________________________________________________________________________________________________
dense (Dense)                   (None, 64)           8256        input_1[0][0]
__________________________________________________________________________________________________
dense_1 (Dense)                 (None, 64)           8256        input_2[0][0]
__________________________________________________________________________________________________
dense_2 (Dense)                 (None, 64)           8256        input_1[0][0]
__________________________________________________________________________________________________
dense_3 (Dense)                 (None, 64)           8256        input_2[0][0]
__________________________________________________________________________________________________
multiply (Multiply)             (None, 64)           0           dense[0][0]
                                                                 dense_1[0][0]
__________________________________________________________________________________________________
multiply_1 (Multiply)           (None, 64)           0           dense_2[0][0]
                                                                 dense_3[0][0]
__________________________________________________________________________________________________
ctr_prob (Dense)                (None, 1)            65          multiply[0][0]
__________________________________________________________________________________________________
cvr_prob (Dense)                (None, 1)            65          multiply_1[0][0]
__________________________________________________________________________________________________
ctcvr_prob (Multiply)           (None, 1)            0           ctr_prob[0][0]
                                                                 cvr_prob[0][0]
==================================================================================================
Total params: 33,154
Trainable params: 33,154
Non-trainable params: 0
__________________________________________________________________________________________________

参考

1. https://mp.weixin.qq.com/s/J3GH1G1xuzEaPdxLzKDGOA