├── .gitattributes ├── NeuRec.ini ├── README.md ├── conf └── SGL.ini ├── data ├── __init__.py ├── dataset.py └── sampler.py ├── dataset ├── amazon-book │ ├── amazon-book.test │ └── amazon-book.train ├── ifashion │ ├── ifashion.test │ └── ifashion.train └── yelp2018 │ ├── yelp2018.test │ └── yelp2018.train ├── local_compile_setup.py ├── log └── amazon-book │ └── SGL │ └── aug_type=ED_reg=1e-3_embed_size=64_n_layers=3_ssl_reg=0.5_ssl_ratio=0.1_ssl_temp=0.2_ssl_mode=both_side_lr=0.001_learner=adam_adj_type=pre_epochs=1000_1649783819.50794291.log ├── main.py ├── model ├── base │ ├── __init__.py │ └── abstract_recommender.py └── general_recommender │ └── SGL.py ├── reckit ├── __init__.py ├── configurator.py ├── cython │ ├── __init__.py │ ├── eval_matrix.pyx │ ├── include │ │ ├── evaluate.h │ │ ├── metric.h │ │ ├── randint.h │ │ ├── sort.h │ │ └── thread_pool.h │ ├── random_choice.pyx │ ├── sort.pyx │ └── tools.pyx ├── dataiterator.py ├── evaluator.py ├── logger.py ├── plot │ ├── __init__.py │ └── style.py ├── preprocessor.py ├── random.py ├── sort.py └── util │ ├── __init__.py │ ├── decorators.py │ └── tools.py ├── requirements.txt └── util ├── common ├── __init__.py └── tool.py ├── pytorch ├── __init__.py ├── func.py └── loss.py └── tensorflow ├── __init__.py ├── func.py └── loss.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/.gitattributes -------------------------------------------------------------------------------- /NeuRec.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/NeuRec.ini -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/README.md -------------------------------------------------------------------------------- /conf/SGL.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/conf/SGL.ini -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/data/dataset.py -------------------------------------------------------------------------------- /data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/data/sampler.py -------------------------------------------------------------------------------- /dataset/amazon-book/amazon-book.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/amazon-book/amazon-book.test -------------------------------------------------------------------------------- /dataset/amazon-book/amazon-book.train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/amazon-book/amazon-book.train -------------------------------------------------------------------------------- /dataset/ifashion/ifashion.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/ifashion/ifashion.test -------------------------------------------------------------------------------- /dataset/ifashion/ifashion.train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/ifashion/ifashion.train -------------------------------------------------------------------------------- /dataset/yelp2018/yelp2018.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/yelp2018/yelp2018.test -------------------------------------------------------------------------------- /dataset/yelp2018/yelp2018.train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/dataset/yelp2018/yelp2018.train -------------------------------------------------------------------------------- /local_compile_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/local_compile_setup.py -------------------------------------------------------------------------------- /log/amazon-book/SGL/aug_type=ED_reg=1e-3_embed_size=64_n_layers=3_ssl_reg=0.5_ssl_ratio=0.1_ssl_temp=0.2_ssl_mode=both_side_lr=0.001_learner=adam_adj_type=pre_epochs=1000_1649783819.50794291.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/log/amazon-book/SGL/aug_type=ED_reg=1e-3_embed_size=64_n_layers=3_ssl_reg=0.5_ssl_ratio=0.1_ssl_temp=0.2_ssl_mode=both_side_lr=0.001_learner=adam_adj_type=pre_epochs=1000_1649783819.50794291.log -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/main.py -------------------------------------------------------------------------------- /model/base/__init__.py: -------------------------------------------------------------------------------- 1 | from .abstract_recommender import * 2 | -------------------------------------------------------------------------------- /model/base/abstract_recommender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/model/base/abstract_recommender.py -------------------------------------------------------------------------------- /model/general_recommender/SGL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/model/general_recommender/SGL.py -------------------------------------------------------------------------------- /reckit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/__init__.py -------------------------------------------------------------------------------- /reckit/configurator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/configurator.py -------------------------------------------------------------------------------- /reckit/cython/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/__init__.py -------------------------------------------------------------------------------- /reckit/cython/eval_matrix.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/eval_matrix.pyx -------------------------------------------------------------------------------- /reckit/cython/include/evaluate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/include/evaluate.h -------------------------------------------------------------------------------- /reckit/cython/include/metric.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/include/metric.h -------------------------------------------------------------------------------- /reckit/cython/include/randint.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/include/randint.h -------------------------------------------------------------------------------- /reckit/cython/include/sort.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/include/sort.h -------------------------------------------------------------------------------- /reckit/cython/include/thread_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/include/thread_pool.h -------------------------------------------------------------------------------- /reckit/cython/random_choice.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/random_choice.pyx -------------------------------------------------------------------------------- /reckit/cython/sort.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/sort.pyx -------------------------------------------------------------------------------- /reckit/cython/tools.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/cython/tools.pyx -------------------------------------------------------------------------------- /reckit/dataiterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/dataiterator.py -------------------------------------------------------------------------------- /reckit/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/evaluator.py -------------------------------------------------------------------------------- /reckit/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/logger.py -------------------------------------------------------------------------------- /reckit/plot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/plot/__init__.py -------------------------------------------------------------------------------- /reckit/plot/style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/plot/style.py -------------------------------------------------------------------------------- /reckit/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/preprocessor.py -------------------------------------------------------------------------------- /reckit/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/random.py -------------------------------------------------------------------------------- /reckit/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/sort.py -------------------------------------------------------------------------------- /reckit/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/util/__init__.py -------------------------------------------------------------------------------- /reckit/util/decorators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/util/decorators.py -------------------------------------------------------------------------------- /reckit/util/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/reckit/util/tools.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/requirements.txt -------------------------------------------------------------------------------- /util/common/__init__.py: -------------------------------------------------------------------------------- 1 | from .tool import * 2 | -------------------------------------------------------------------------------- /util/common/tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/common/tool.py -------------------------------------------------------------------------------- /util/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/pytorch/__init__.py -------------------------------------------------------------------------------- /util/pytorch/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/pytorch/func.py -------------------------------------------------------------------------------- /util/pytorch/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/pytorch/loss.py -------------------------------------------------------------------------------- /util/tensorflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/tensorflow/__init__.py -------------------------------------------------------------------------------- /util/tensorflow/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/tensorflow/func.py -------------------------------------------------------------------------------- /util/tensorflow/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wujcan/SGL-Torch/HEAD/util/tensorflow/loss.py --------------------------------------------------------------------------------