/CLEA

Primary LanguagePython

CLEA

The World is Binary: Contrastive Learning for Denoising Next Basket Recommendation


Datasets

for example:
{
u_0: [ [ i_0, i_1 ], [ i_2, i_3, i_4]] ,
u_1: [ [ i_5, i_6], [ i_7, i_8], [ i_9] ]
}

Note!!! user_id and item_id begin with 0

Requirements

  • python 3.7
  • pytorch 1.7.0

Run

1.Instacart

sh Instacart_0.sh

2.Dunnhumby

sh Dunn_0.sh