├── README.md ├── RQVAE ├── .DS_Store ├── datasets.py ├── main.py ├── models │ ├── .DS_Store │ ├── layers.py │ ├── rq.py │ ├── rqvae.py │ └── vq.py ├── run_pretrain.sh ├── trainer.py └── utils.py ├── accelerate_config_ddp.yaml ├── asset └── model.png ├── config └── scientific.yaml ├── data.py ├── layers.py ├── main.py ├── metrics.py ├── model.py ├── run.sh ├── trainer.py ├── utils.py └── vq.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/README.md -------------------------------------------------------------------------------- /RQVAE/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/.DS_Store -------------------------------------------------------------------------------- /RQVAE/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/datasets.py -------------------------------------------------------------------------------- /RQVAE/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/main.py -------------------------------------------------------------------------------- /RQVAE/models/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/models/.DS_Store -------------------------------------------------------------------------------- /RQVAE/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/models/layers.py -------------------------------------------------------------------------------- /RQVAE/models/rq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/models/rq.py -------------------------------------------------------------------------------- /RQVAE/models/rqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/models/rqvae.py -------------------------------------------------------------------------------- /RQVAE/models/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/models/vq.py -------------------------------------------------------------------------------- /RQVAE/run_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/run_pretrain.sh -------------------------------------------------------------------------------- /RQVAE/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/trainer.py -------------------------------------------------------------------------------- /RQVAE/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/RQVAE/utils.py -------------------------------------------------------------------------------- /accelerate_config_ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/accelerate_config_ddp.yaml -------------------------------------------------------------------------------- /asset/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/asset/model.png -------------------------------------------------------------------------------- /config/scientific.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/config/scientific.yaml -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/data.py -------------------------------------------------------------------------------- /layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/layers.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/main.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/metrics.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/model.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/run.sh -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/utils.py -------------------------------------------------------------------------------- /vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCAIBox/ETEGRec/HEAD/vq.py --------------------------------------------------------------------------------