最近编写程序的时候发现了下面的这样一个错误:
后来发现是结果的标签出现了错误,在model之中本身定义的循环内容如下:
def cycle(self,random=True):
while True:
for d in self.__iter__(random):
print('d = ')
print(d)
yield d
这里输出的d的内容为
报错的原因在于这里的标签只有一个相应的数值,与d中的输入数值不匹配,d中有多个数值,应该匹配上多个相应的标签。
注意加上标签之后,不能以数组的形式返回,而应该以np.array()的形式返回,比如返回batch_label_ids,则返回的内容应该为np.array(batch_label_ids)