我正在嘗試學習和理解如何使用 ANN 實作多類分類。就我而言,我有 16 個類(0-15),我的標簽資料集包含一列帶有標簽值。所以我知道輸出層應該有與類相同數量的神經元。當我創建具有 16 個神經元的輸出層時,我收到以下錯誤訊息:
形狀 (32, 1) 和 (32, 16) 不兼容
我遵循了類似問題的解決方案:
y_train= tf.one_hot(y_train, 16)
我收到以下錯誤:
ValueError:形狀 (32, 1, 16) 和 (32, 16) 不兼容
我知道問題出在標簽的形狀上,但我不知道如何解決。
感謝您提供的任何幫助。
uj5u.com熱心網友回復:
解決方案是使用以下內容:
from tensorflow.keras.utils import to_categorical
y_cat_test = to_categorical(y_test,16)
轉載請註明出處,本文鏈接:https://www.uj5u.com/qukuanlian/473772.html
