search

chenxijun1029 / DeepFM_with_PyTorch

A PyTorch implementation of DeepFM for CTR prediction problem.
379 stars 100 forks source link

20 #4

Open chrisyxue opened 5 years ago

chrisyxue commented 5 years ago

Hello, Could you tell me why it is 20? What's the meaning of it? In the part of DeepFM.Forward """ emb = self.fm_first_order_embeddings[20] print(Xi.size()) for num in Xi[:, 20, :][0]: if num > self.feature_sizes[20]: print("index out") """

weihuiHong commented 5 years ago

This code doesn't work for DeepFM. I think the author just wants to output intermediate values for viewing.

chenxijun1029 commented 5 years ago

As weihuiHong said, the code was just for test