You can email me if you have any questions about the performance and efficiency of CpRec.
you can run nextitrec_eval.py (python nextitrec_eval.py) directly, which includes training and testing.
Each training epoch, test once test set.
--softmax_type: This is used to control whether block-wise embedding is used. There are 5 classes:
FullSoftmax：The input layer and output layer use basic embedding and softmax respectively.
Block_Input_Full：Only the input layer uses block-wise embedding, and the output layer uses basic softmax.
Block_for_Softmax：Only the output layer uses block-wise embedding, and the output layer uses basic softmax.
Block_Input_Softmax：Both the input layer and the output layer use block-wise embedding.
Block_Input_Softmax_Inference： Both the input layer and the output layer use block-wise embedding, but fast inference is used in the prediction stage. it should be noted that the use of this type requires batch_size_test = 1
--use_parametersharing_type: This is used to control whether to use layer-wise parameter sharing, there are 5 classes:
original：No parameter sharing.
cross-layer：use the cross-layer parameter sharing.
cross-block：use the cross-block parameter sharing.
adjacent-layer：use the adjacent-layer parameter sharing.
adjacent-block：use the adjacent-block parameter sharing.
you should cross-valid the number of blocks and the embedding size of each block, but we strongly suggest you use 3~5 blocks.
--use_embedding_type_factor is the attenuation factor of block dimension in block-wise embedding, the default is 4. You can also manually set the dimension of each block, e.g. [300, 200, 100].
you can download a large sequential dataset of movielen that has been pre-processed: https://pan.baidu.com/s/1XZCUoRwWFa8fpRWpEA2HPQ
code (提取码):fo51
https://github.com/fajieyuan/nextitnet code that including Caser and GRURec
https://github.com/graytowne/caser code
A Generic Network Compression Framework for Sequential Recommender Systems
@article{sun2020generic,
title={A Generic Network Compression Framework for Sequential Recommender Systems},
author={Sun, Yang and Yuan, Fajie and Yang, Min and Wei, Guoao and Zhao, Zhou and Liu, Duo},
journal={arXiv preprint arXiv:2004.13139},
year={2020}
}
Please cite this paper if you find our code is useful.