from DIAN.data import Data
from DIAN.models.BPR import BPR
data = Data(train_file='data/ml-1m/'+'train_users.dat', test_file='data/ml-1m/'+'test_users.dat')
model = BPR(data, emb_dim=16, batch_size=1024)
model.train(200, lr=0.01, optimizer='Adam')
model.predict()
Performance
Model HitRatio@20 MAP@20
BPR 0.20 0.06
FM 0.18 0.04