def init_lambda_network(self):
assert hasattr(self, 'lambda_network')
if self.use_cuda:
use_cuda(True, self._opt['device_id'])
self.lambda_network.cuda()
self.optimizer = use_optimizer(self.lambda_network,
self.lambda_network_opt)
self.scheduler = ExponentialLR(
self.optimizer, gamma=self.lambda_network_opt['lr_exp_decay'])
def init_episode(self):
opt = self.opt
self.model = MF(opt)
self._train_step_idx = 0
if self.use_cuda:
use_cuda(True, opt['device_id'])
self.model.cuda()
self.optimizer = use_optimizer(self.model, opt)
self.scheduler = ExponentialLR(self.optimizer,
gamma=opt['lr_exp_decay'])
self.param = [p.data.clone() for p in self.model.parameters()]
def __init__(self, opt):
super(MFBPRFactorizer, self).__init__(opt)
self.model = MF(opt)
self.opt = opt
if self.use_cuda:
use_cuda(True, opt['device_id'])
self.model.cuda()
self.optimizer = use_optimizer(self.model, opt)
self.scheduler = ExponentialLR(self.optimizer,
gamma=opt['lr_exp_decay'])
def init_episode(self):
opt = self.opt
if opt['model'] == 'linear':
self.model = LR(opt)
elif opt['model'] == 'fm':
self.model = FM(opt)
elif opt['model'] == 'deepfm':
self.model = DeepFM(opt)
elif opt['model'] == 'autoint':
self.model = AutoInt(opt)
else:
raise ValueError("Invalid FM model type: {}".format(opt['model']))
self._train_step_idx = 0
if self.use_cuda:
use_cuda(True, opt['device_id'])
self.model.cuda()
self.optimizer = use_optimizer(self.model, opt)
self.scheduler = ExponentialLR(self.optimizer,
gamma=opt['lr_exp_decay'])
def __init__(self, opt):
super(FMFactorizer, self).__init__(opt)
self.opt = opt
if opt['model'] == 'linear':
self.model = LR(opt)
elif opt['model'] == 'fm':
self.model = FM(opt)
elif opt['model'] == 'deepfm':
self.model = DeepFM(opt)
elif opt['model'] == 'autoint':
self.model = AutoInt(opt)
else:
raise ValueError("Invalid FM model type: {}".format(opt['model']))
if self.use_cuda:
use_cuda(True, opt['device_id'])
self.model.cuda()
self.optimizer = use_optimizer(self.model, opt)
self.scheduler = ExponentialLR(self.optimizer,
gamma=opt['lr_exp_decay'])