发现我的代码
python
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处理这种情况
python
if self.flag == 'valid':
us = self.user_valid_list[idx]
elif self.flag == 'test':
us = self.user_test_list[idx]
else:
us = 0
info = sample(us)