├── .gitignore ├── README.md ├── configs ├── adressa.yaml ├── dataset │ ├── AdressaSmall.yaml │ ├── MINDLarge.yaml │ ├── MINDSmall.yaml │ └── default.yaml ├── default.yaml ├── large.yaml ├── logger │ └── wandb.yaml ├── model │ ├── GLORY.yaml │ └── default.yaml ├── optimizer │ └── adam.yaml ├── path │ └── default.yaml └── small.yaml ├── glory.jpg ├── requirements.txt ├── scripts └── data_download.sh └── src ├── dataload ├── data_load.py ├── data_preprocess.py └── dataset.py ├── main.py ├── models ├── GLORY.py ├── base │ ├── function.py │ └── layers.py └── component │ ├── candidate_encoder.py │ ├── click_encoder.py │ ├── entity_encoder.py │ ├── nce_loss.py │ ├── news_encoder.py │ └── user_encoder.py └── utils ├── common.py └── metrics.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/README.md -------------------------------------------------------------------------------- /configs/adressa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/adressa.yaml -------------------------------------------------------------------------------- /configs/dataset/AdressaSmall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/dataset/AdressaSmall.yaml -------------------------------------------------------------------------------- /configs/dataset/MINDLarge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/dataset/MINDLarge.yaml -------------------------------------------------------------------------------- /configs/dataset/MINDSmall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/dataset/MINDSmall.yaml -------------------------------------------------------------------------------- /configs/dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/dataset/default.yaml -------------------------------------------------------------------------------- /configs/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/default.yaml -------------------------------------------------------------------------------- /configs/large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/large.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/GLORY.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/model/GLORY.yaml -------------------------------------------------------------------------------- /configs/model/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/model/default.yaml -------------------------------------------------------------------------------- /configs/optimizer/adam.yaml: -------------------------------------------------------------------------------- 1 | lr: 0.0001 -------------------------------------------------------------------------------- /configs/path/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/path/default.yaml -------------------------------------------------------------------------------- /configs/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/configs/small.yaml -------------------------------------------------------------------------------- /glory.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/glory.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/scripts/data_download.sh -------------------------------------------------------------------------------- /src/dataload/data_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/dataload/data_load.py -------------------------------------------------------------------------------- /src/dataload/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/dataload/data_preprocess.py -------------------------------------------------------------------------------- /src/dataload/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/dataload/dataset.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/main.py -------------------------------------------------------------------------------- /src/models/GLORY.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/GLORY.py -------------------------------------------------------------------------------- /src/models/base/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/base/function.py -------------------------------------------------------------------------------- /src/models/base/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/base/layers.py -------------------------------------------------------------------------------- /src/models/component/candidate_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/candidate_encoder.py -------------------------------------------------------------------------------- /src/models/component/click_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/click_encoder.py -------------------------------------------------------------------------------- /src/models/component/entity_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/entity_encoder.py -------------------------------------------------------------------------------- /src/models/component/nce_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/nce_loss.py -------------------------------------------------------------------------------- /src/models/component/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/news_encoder.py -------------------------------------------------------------------------------- /src/models/component/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/models/component/user_encoder.py -------------------------------------------------------------------------------- /src/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/utils/common.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyrolls/GLORY/HEAD/src/utils/metrics.py --------------------------------------------------------------------------------