Add the BST model and its demo on MovieLens dataset

meta-soul / MetaSpore

A unified end-to-end machine intelligence platform

Apache License 2.0

642 stars 118 forks source link

Add the BST model and its demo on MovieLens dataset #210

Closed Cheng-Su closed 2 years ago

Cheng-Su commented 2 years ago

Implement the Behavior Sequence Transformer model. The performance of this model on MovieLens dataset is as follows, which should be improve in the future.

Dataset	ml-1m	ml-25m
train AUC	0.767	0.891
test AUC	0.725	0.857

close #211