文章目录
一些犯过的愚愚蠢蠢的错误(1)
背景介绍
统计每个label的index存在digit_indices中,如果有5个label,那么digit_indices应该是有五个list元素的list
- data 是一个list
- label是与其对应的list(0,1,2,3.)
Bug
digit_indices = [np.where(label == i)[0] for i in range(num_classes)]
没有任何error,只是在运行过程中发现digit_indices是空list,诡异。
探索中
看上去都是int为什么label==i 的判断后的list是空?(黑人问号),这个时候我还是没有想到int32和int64的区别。
- 只是尝试写成:
digit_indices = [np.where(label == i)[0] for i in np.unique(label)]
这次的digit_indices不是空的了,但是为什么呢?
- 改成这样试试:
digit_indices = [np.where(label == i)[0] for i in [0,1]]
空的 - 灵光了,意识到是type的问题,label的int是int64,但是for循环中的i是int32,等不了
digit_indices = [np.where(label == i)[0] for i in np.array([0,1],dtype=int64)]
哈哈哈,被自己刚睡醒的愚蠢笑掉大牙,dtype哪里有int64的说法??
上正经的!
digit_indices = [np.where(label == i)[0] for i in np.array([0,1],dtype=int)]