def get_feed_dict(self, corpus, data, batch_start, batch_size, phase):
feed_dict = BaseModel.get_feed_dict(self, corpus, data, batch_start,
batch_size, phase)
real_batch_size = feed_dict['batch_size']
user_ids = data['user_id'][batch_start:batch_start +
real_batch_size].values
feed_dict['user_id'] = utils.numpy_to_torch(user_ids) # [batch_size]
return feed_dict
def get_feed_dict(self, corpus, data, batch_start, batch_size, phase):
feed_dict = BaseModel.get_feed_dict(self, corpus, data, batch_start,
batch_size, phase)
real_batch_size = feed_dict['batch_size']
history_items = data['item_his'][batch_start:batch_start +
real_batch_size].values
lengths = data['his_length'][batch_start:batch_start +
real_batch_size].values
feed_dict['history_items'] = utils.numpy_to_torch(
utils.pad_lst(history_items))
feed_dict['lengths'] = utils.numpy_to_torch(lengths)
return feed_dict