SIAN
1.0.0
ECML-PKDD 논문 "친구 강화 추천을 위한 사회적 영향력 주의 신경망"의 코드 및 데이터
├── baselines # baseline code
│ ├── Eval4Baselines.py
├── code # Our Model: SIAN
│ ├── Attention.py # attention layer
│ ├── DataUtil.py # data loader
│ ├── Evaluation.py # model evaluation
│ ├── FeatureAgg.py # attentive feature aggregator
│ ├── Fusion.py # feature fusion layer
│ ├── HeteInf.py # the main class for SIAN
│ ├── InfluenceProp.py # social influence coupler
│ ├── Logging.py #log
│ └── trainHeteInf.py # the main function for SIAN
└── data # dataset
├── Data4Baselines.ipynb #
├── DataProcessor.ipynb
├── ItemProfileEmbed.ipynb
├── WechatTencent.ipynb
├── wxt # FWD dataset
│ ├── biz2id
│ ├── biz_profile.npy
│ ├── item2id
│ ├── item_profile.npy
│ ├── user2id
│ ├── user_profile.npy
│ ├── wxt.att.analysis
│ ├── wxt.interaction.graph
│ ├── wxt.item.biz
│ ├── wxt.social.graph
│ ├── wxt.test.rating.712
│ ├── wxt.train.rating.712
│ ├── wxt.user.biz
│ └── wxt.val.rating.712
├── wxt.ipynb
├── yelp # yelp dataset
│ ├── item_profile.npy
│ ├── user_profile.npy
│ ├── yelp.att.analysis
│ ├── yelp.interaction.graph
│ ├── yelp.social.graph
│ ├── yelp.test.rating.712
│ ├── yelp.train.rating.712
│ └── yelp.val.rating.712
├── yelp.ipynb
├── log # saved log file
│ ├── wxt.0.0.6023.0.35225.model
FWD 데이터 세트(예: wxt 데이터)는 Google Drive 및 BaiduYun(提取码:i6qy)에서 다운로드할 수 있습니다.
log/
dir에 저장된 모델을 로드합니다.
python trainHeteInf.py --help
@inproceedings{Yuanfu2020SIAN,
title={Social Influence Attentive Neural Network for Friend-Enhanced Recommendation},
author={Yuanfu Lu, Ruobing Xie, Chuan Shi, Yuan Fang, Wei Zhang, Xu Zhang, Leyu Lin.}
booktitle={Proceedings of ECML-PKDD},
year={2020}
}