├── EasyRec.png ├── README.md ├── cf_rec ├── __init__.py ├── config │ ├── configurator.py │ └── modelconf │ │ ├── default.yml │ │ ├── gccf.yml │ │ ├── gccf_plus.yml │ │ ├── lightgcn.yml │ │ └── lightgcn_plus.yml ├── data_utils │ ├── __init__.py │ ├── build_data_handler.py │ ├── data_handler_general_cf.py │ └── datasets_general_cf.py ├── models │ ├── __init__.py │ ├── aug_utils.py │ ├── base_model.py │ ├── bulid_model.py │ ├── general_cf │ │ ├── gccf.py │ │ ├── gccf_plus.py │ │ ├── lightgcn.py │ │ └── lightgcn_plus.py │ ├── loss_utils.py │ └── model_utils.py ├── run.py └── trainer │ ├── __init__.py │ ├── build_trainer.py │ ├── logger.py │ ├── metrics.py │ ├── trainer.py │ └── utils.py ├── create_roberta_small.py ├── encode_easyrec.py ├── eval_text_emb.py ├── generation ├── diverse_profile.py ├── generate_profile.py └── instruction │ ├── item_system_prompt.txt │ ├── item_system_prompt_diverse.txt │ ├── user_system_prompt.txt │ └── user_system_prompt_diverse.txt ├── model.py ├── train_base.sh ├── train_easyrec.py ├── train_large.sh ├── train_small.sh └── utility ├── load_data.py ├── logger.py ├── loss_utils.py ├── metric.py └── trainer.py /EasyRec.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/EasyRec.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/README.md -------------------------------------------------------------------------------- /cf_rec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cf_rec/config/configurator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/configurator.py -------------------------------------------------------------------------------- /cf_rec/config/modelconf/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/modelconf/default.yml -------------------------------------------------------------------------------- /cf_rec/config/modelconf/gccf.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/modelconf/gccf.yml -------------------------------------------------------------------------------- /cf_rec/config/modelconf/gccf_plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/modelconf/gccf_plus.yml -------------------------------------------------------------------------------- /cf_rec/config/modelconf/lightgcn.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/modelconf/lightgcn.yml -------------------------------------------------------------------------------- /cf_rec/config/modelconf/lightgcn_plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/config/modelconf/lightgcn_plus.yml -------------------------------------------------------------------------------- /cf_rec/data_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/data_utils/__init__.py -------------------------------------------------------------------------------- /cf_rec/data_utils/build_data_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/data_utils/build_data_handler.py -------------------------------------------------------------------------------- /cf_rec/data_utils/data_handler_general_cf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/data_utils/data_handler_general_cf.py -------------------------------------------------------------------------------- /cf_rec/data_utils/datasets_general_cf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/data_utils/datasets_general_cf.py -------------------------------------------------------------------------------- /cf_rec/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cf_rec/models/aug_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/aug_utils.py -------------------------------------------------------------------------------- /cf_rec/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/base_model.py -------------------------------------------------------------------------------- /cf_rec/models/bulid_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/bulid_model.py -------------------------------------------------------------------------------- /cf_rec/models/general_cf/gccf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/general_cf/gccf.py -------------------------------------------------------------------------------- /cf_rec/models/general_cf/gccf_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/general_cf/gccf_plus.py -------------------------------------------------------------------------------- /cf_rec/models/general_cf/lightgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/general_cf/lightgcn.py -------------------------------------------------------------------------------- /cf_rec/models/general_cf/lightgcn_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/general_cf/lightgcn_plus.py -------------------------------------------------------------------------------- /cf_rec/models/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/loss_utils.py -------------------------------------------------------------------------------- /cf_rec/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/models/model_utils.py -------------------------------------------------------------------------------- /cf_rec/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/run.py -------------------------------------------------------------------------------- /cf_rec/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/__init__.py -------------------------------------------------------------------------------- /cf_rec/trainer/build_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/build_trainer.py -------------------------------------------------------------------------------- /cf_rec/trainer/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/logger.py -------------------------------------------------------------------------------- /cf_rec/trainer/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/metrics.py -------------------------------------------------------------------------------- /cf_rec/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/trainer.py -------------------------------------------------------------------------------- /cf_rec/trainer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/cf_rec/trainer/utils.py -------------------------------------------------------------------------------- /create_roberta_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/create_roberta_small.py -------------------------------------------------------------------------------- /encode_easyrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/encode_easyrec.py -------------------------------------------------------------------------------- /eval_text_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/eval_text_emb.py -------------------------------------------------------------------------------- /generation/diverse_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/diverse_profile.py -------------------------------------------------------------------------------- /generation/generate_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/generate_profile.py -------------------------------------------------------------------------------- /generation/instruction/item_system_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/instruction/item_system_prompt.txt -------------------------------------------------------------------------------- /generation/instruction/item_system_prompt_diverse.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/instruction/item_system_prompt_diverse.txt -------------------------------------------------------------------------------- /generation/instruction/user_system_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/instruction/user_system_prompt.txt -------------------------------------------------------------------------------- /generation/instruction/user_system_prompt_diverse.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/generation/instruction/user_system_prompt_diverse.txt -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/model.py -------------------------------------------------------------------------------- /train_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/train_base.sh -------------------------------------------------------------------------------- /train_easyrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/train_easyrec.py -------------------------------------------------------------------------------- /train_large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/train_large.sh -------------------------------------------------------------------------------- /train_small.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/train_small.sh -------------------------------------------------------------------------------- /utility/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/utility/load_data.py -------------------------------------------------------------------------------- /utility/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/utility/logger.py -------------------------------------------------------------------------------- /utility/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/utility/loss_utils.py -------------------------------------------------------------------------------- /utility/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/utility/metric.py -------------------------------------------------------------------------------- /utility/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/EasyRec/HEAD/utility/trainer.py --------------------------------------------------------------------------------