def test05_validation(self): np.random.seed(7) opt = BPRMFOption().get_default_option() opt.d = 5 opt.num_workers = 4 opt.num_iters = 500 opt.random_seed = 7 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)
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)