I rebuild the algorithm from the "Attention is all you need"-Paper and try to use attention to predict which item comes next in a Market Basket Optimisation dataset from kaggle (https://www.kaggle.com/roshansharma/market-basket-optimization).
lschmiddey/BERT
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Sequence prediction with BERT
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published