Johnson0722 / CTR_Prediction

CTR prediction using FM FFM and DeepFM
746 stars 297 forks source link

FM #20

Open SPY-Ming opened 3 years ago

SPY-Ming commented 3 years ago

各位好,我flow这份代码,在FM模型中(FFM还没来及看),sparse_data_fraction作为数据输入,在train_model函数中被分成若干batch。每一个batch是[batch_size*21, 2]的格式。但是在feed_dict,作者写了三个东西(batch_indexes, batch_values, batch_shape),但是他的placeholder形状为[None, feature_size]。请问这是怎么回事?他真实的placeholder值应该是多少?另外这份代码的one-hot embedding处其实没有做embedding,请问后边one-hot embedding是怎么做的?我没有找到相关的函数