qianrenjian / frnet

The Source Code of FRNet

Enhancing CTR Prediction with Context-Aware Feature Representation Learning

This is our implement for FRNet

How to run the code.

1. Download corresponding datasets.

   • Criteo
   • Frappe
   • Ml-tag
   • Malware

2. For example, put the Frappe or Ml-tag data in the corresponding folder.

3. Run the code with GPU:

   CUDA_VISIBLE_DEVICES=0 python main_ml_base.py --choice 0

You can cite this article:

Fangye Wang, Yingxu Wang, Dongsheng Li, Hansu Gu, Tun Lu, Peng Zhang, and Ning Gu. 2022. Enhancing CTR Prediction with Context-Aware Feature Representation Learning. In Proceedings of the 45th International ACM SIGIR Conference on Research and Development in Information Retrieval (SIGIR ’22). 343–352.

@inproceedings{wang2022frnet,
author = {Wang, Fangye and Wang, Yingxu and Li, Dongsheng and Gu, Hansu and Lu, Tun and Zhang, Peng and Gu, Ning},
title = {Enhancing CTR Prediction with Context-Aware Feature Representation Learning},
booktitle={Proceedings of the 45th International ACM SIGIR Conference on Research and Development in Information Retrieval},
year = {2022},
pages = {343–352},
series = {SIGIR '22}
}