BERT4Rec runable with Tensorflow 2.x
Article link: BERT4Rec https://arxiv.org/pdf/1904.06690.pdf
Base repository: https://github.com/FeiSun/BERT4Rec
Tensorflow 2.x (2.5)
Prepare train data: data_processing.py
Train and Evaluation: bert4rec_main.py
-
Notifications
You must be signed in to change notification settings - Fork 4
tunghia1890/BERT4Rec_TF2x
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
BERT4Rec runable with Tensorflow 2.x
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published