Sequence prediction with BERT

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).