A TensorFlow implementation for C2-Rec
This is a TensorFlow implementation for C2-Rec We refer to the repo SASRec. Requirements requirement.txt Datasets This repo includes Amazon Beauty dataset as an example. you could also download Amazon review data from here. Model Training To train model on Beauty: python main.py –dataset=Beauty –train_dir=model_train –maxlen=50 –dropout_rate=0.5 –con_alpha=5.0 –rd_alpha=1.0 –neg_test=500 –user_reg_type=kl –lr=0.001 –rd_reduce=mean –neg_sample_n=50 GitHub View Github
Read more