2022 iThome 鐵人賽
分享至
廢廢今天參考了一篇超有用的文章np_utils.to_categorical函數裡面介紹了該函數的參數。之前用np_utils.to_categorical(train_label, num_classes=5),因為只有5個label所以num_classes設為5,結果一直出錯說超過長度,後來就一直使用10。但是這篇文說明了當label=[0,1,2]時,num_classes=3,所以label設定要從0開始,之前廢廢把label設成1~5所以出現錯誤,要改成0~4。
IT邦幫忙