Merge pull request #713 from cyLi-Tiger/0.2.x

FEA：Add NNCF
RUCAIBox · Feb 24, 2021 · fcb97f1 · fcb97f1
2 parents 684d259 + 744a905
commit fcb97f1
Show file tree

Hide file tree

Showing 6 changed files with 426 additions and 0 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -24,6 +24,9 @@ jobs:
         pip install pytest
         pip install dgl
         pip install xgboost
+        pip install community
+        pip install networkx
+        pip install python-louvain
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
         
     # Use "python -m pytest" instead of "pytest" to fix imports

diff --git a/recbole/model/general_recommender/__init__.py b/recbole/model/general_recommender/__init__.py
@@ -17,3 +17,4 @@
 from recbole.model.general_recommender.ngcf import NGCF
 from recbole.model.general_recommender.pop import Pop
 from recbole.model.general_recommender.spectralcf import SpectralCF
+from recbole.model.general_recommender.nncf import NNCF