1. 程式人生 > >[Bug集合]Cast string to int32 is not supported

[Bug集合]Cast string to int32 is not supported

出現於我想打亂訓練集輸入順序的時候。
如:

temp = np.array([image_list, label_list])
temp = temp.transpose()
np.random.shuffle(temp)

all_image_list = temp[:, 0]
all_label_list = temp[:, 1]

原來的影象list和標籤list合前,標籤是int32型,結果拆開後,標籤list變成了str型,然後它就做不了神經網路中的輸出層內容了。
送到模型訓練之後就會報錯。
所以正常執行程式需要把它改回int32型。

all_label_list =
[int(float(i)) for i in all_label_list]