/CSCNN

CSCNN Unofficial Implementation: Category-Specific CNN for Visual-aware CTR Prediction at JD.com

Primary LanguagePython

CSCNN

CSCNN Unofficial Implementation: Category-Specific CNN for Visual-aware CTR Prediction at JD.com. KDD 2020. paper link: https://dl.acm.org/doi/abs/10.1145/3394486.3403319

main.py

You can see the "CSCNN" funtion in main.py to figure out the usage of CSCNN. The code is based on Resnet18 (only "resblock" contains CSCNN in main.py). Before you use the function, please prepare the embeddings for target items, which is explained in the paper.

CSCNN.py

Unofficial implementation of CSCNN, based on CBAM attention module.