├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── bert_train ├── bert_config_beauty_128.json ├── bert_config_beauty_256.json ├── bert_config_beauty_64.json ├── bert_config_ml-1m_128.json ├── bert_config_ml-1m_256.json ├── bert_config_ml-1m_64.json ├── bert_config_ml-20m_128.json ├── bert_config_ml-20m_256.json ├── bert_config_ml-20m_64.json ├── bert_config_steam_128.json ├── bert_config_steam_256.json └── bert_config_steam_64.json ├── data ├── beauty.txt ├── ml-1m.txt ├── ml-20m.zip └── steam.txt ├── gen_data.py ├── gen_data_fin.py ├── modeling.py ├── optimization.py ├── run.py ├── run_beauty.sh ├── run_ml-1m.sh ├── run_ml-20m.sh ├── run_steam.sh ├── util.py └── vocab.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/__init__.py -------------------------------------------------------------------------------- /bert_train/bert_config_beauty_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_beauty_128.json -------------------------------------------------------------------------------- /bert_train/bert_config_beauty_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_beauty_256.json -------------------------------------------------------------------------------- /bert_train/bert_config_beauty_64.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_beauty_64.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-1m_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-1m_128.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-1m_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-1m_256.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-1m_64.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-1m_64.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-20m_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-20m_128.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-20m_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-20m_256.json -------------------------------------------------------------------------------- /bert_train/bert_config_ml-20m_64.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_ml-20m_64.json -------------------------------------------------------------------------------- /bert_train/bert_config_steam_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_steam_128.json -------------------------------------------------------------------------------- /bert_train/bert_config_steam_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_steam_256.json -------------------------------------------------------------------------------- /bert_train/bert_config_steam_64.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/bert_train/bert_config_steam_64.json -------------------------------------------------------------------------------- /data/beauty.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/data/beauty.txt -------------------------------------------------------------------------------- /data/ml-1m.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/data/ml-1m.txt -------------------------------------------------------------------------------- /data/ml-20m.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/data/ml-20m.zip -------------------------------------------------------------------------------- /data/steam.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/data/steam.txt -------------------------------------------------------------------------------- /gen_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/gen_data.py -------------------------------------------------------------------------------- /gen_data_fin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/gen_data_fin.py -------------------------------------------------------------------------------- /modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/modeling.py -------------------------------------------------------------------------------- /optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/optimization.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/run.py -------------------------------------------------------------------------------- /run_beauty.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/run_beauty.sh -------------------------------------------------------------------------------- /run_ml-1m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/run_ml-1m.sh -------------------------------------------------------------------------------- /run_ml-20m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/run_ml-20m.sh -------------------------------------------------------------------------------- /run_steam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/run_steam.sh -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/util.py -------------------------------------------------------------------------------- /vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FeiSun/BERT4Rec/HEAD/vocab.py --------------------------------------------------------------------------------