search

ruizhang-ai / SIGN-Detecting-Beneficial-Feature-Interactions-for-Recommender-Systems

Detecting Beneficial Feature Interactions for Recommender Systems, AAAI 2021
MIT License
33 stars 6 forks source link

linkpred模块代码含义 #6

Closed discivigour closed 1 year ago

discivigour commented 1 year ago

请问LinkPred类的forward函数中的torch.log2()函数和 s = torch.sigmoid(sum_log/self.temp) s = s * (self.inter_max - self.inter_min) + self.inter_min 到底是干什么用的呢？

ethanmock commented 1 year ago

Hi,

It maps the predictions to the hard concrete distribution. You can see the description in the Appendix "Approximate L0 Regularization with Hard Concrete Distribution".