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邦幫忙
看影片追技術