Skip to content

Commit

Permalink
fix max item
Browse files Browse the repository at this point in the history
  • Loading branch information
LuckyHorseshoe-chan committed Apr 29, 2024
1 parent 887b100 commit de853ef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ def main():

args.data_file = args.data_dir + 'train_' + args.data_name + '.txt'
train_seq_dic, train_max_item, num_users = get_seq_dic(args)
args.item_size = max_item + 1
args.item_size = train_max_item + 1
args.num_users = num_users + 1
train_dataloader = get_dataloder(args, train_seq_dic, 'train')

Expand Down

0 comments on commit de853ef

Please sign in to comment.