def test12_gpu_train_ml_20m(self):
if not inited_CUBPR:
return
opt = BPRMFOption().get_default_option()
opt.accelerator = True
opt.d = 100
opt.verify_neg = False
opt.num_iters = 30
opt.evaluation_period = 5
opt.validation = aux.Option({'topk': 10})
self._test7_train_ml_20m(BPRMF, opt)
def test11_gpu_validation(self):
if not inited_CUBPR:
return
np.random.seed(7)
opt = BPRMFOption().get_default_option()
opt.d = 100
opt.verify_neg = False
opt.accelerator = True
opt.lr = 0.01
opt.reg_b = 10.0
opt.num_iters = 500
opt.evaluation_period = 50
opt.random_seed = 777
opt.validation = aux.Option({'topk': 10})
opt.tensorboard = aux.Option({'root': './tb', 'name': 'bpr'})
self._test5_validation(BPRMF, opt, ndcg=0.03, map=0.02)