发现我的代码
if self.flag == 'valid':us = self.user_valid_list[idx]
elif self.flag == 'test':us = self.user_test_list[idx]info = sample(us)
如果我的flag不是train和valid中的值,那么就会出现问题,因此再加上一个else处理这种情况
if self.flag == 'valid':us = self.user_valid_list[idx]
elif self.flag == 'test':us = self.user_test_list[idx]
else:us = 0info = sample(us)