unboundlocalerror: local variable ‘××ב referenced before assignment

发现我的代码

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 = 0

info = sample(us)
           

你可能感兴趣的:(小记,机器学习)