add struc2vec code

WDdeBWT · Jun 24, 2020 · 1547bae · 1547bae
1 parent 02e8567
commit 1547bae
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 362 deletions.
diff --git a/README.md b/README.md
@@ -1,18 +1,24 @@
 # RecommendationCTR
 Recommendation model for CTR prediction
 
-## Implementation based on KGAT
+## Implementation based on following project
 
-* https://github.com/LunaBlack/KGAT-pytorch
+* https://github.com/LunaBlack/KGAT-pytorch # model KGAT Pyorch implementation
 
-* https://github.com/xiangwang1223/knowledge_graph_attention_network
+* https://github.com/xiangwang1223/knowledge_graph_attention_network # model KGAT Tensorflow implementation
+
+* https://github.com/gusye1234/LightGCN-PyTorch # model LightGCN Pyorch implementation
+
+* https://github.com/shenweichen/GraphEmbedding # a lot of graph embedding model (DeepWalk, Struc2Vec, etc.) implementation
 
 ## Environment Requirement
 The code has been tested running under Python 3.6.8. The required packages are as follows:
-* torch == 1.3.1
-* dgl-cu90 == 0.4.1
-* numpy == 1.15.4
-* pandas == 0.23.1
-* scipy == 1.1.0
-* sklearn == 0.20.0
-
+* torch
+* dgl-cu90
+* numpy
+* pandas
+* scipy
+* sklearn
+* networkx
+* fastdtw
+* joblib
diff --git a/lgcn_model.py → light_gcn/lgcn_model.py b/lgcn_model.py → light_gcn/lgcn_model.py
diff --git a/script_kgat.py b/script_kgat.py
diff --git a/script_lgcn.py b/script_lgcn.py
@@ -7,7 +7,7 @@
 import numpy as np
 
 from cf_dataset import DataOnlyCF
-from lgcn_model import LightGCN
+from light_gcn.lgcn_model import LightGCN
 from metrics import precision_and_recall, ndcg, auc
 
 EPOCH = 200

diff --git a/test_dgl.py b/test_dgl.py