我正在嘗試使用 AlexNet 測(cè)試數(shù)據(jù)集。我正在使用這個(gè)實(shí)現(xiàn)。問(wèn)題是我的數(shù)據(jù)集有 11 個(gè)類，但該站點(diǎn)上的代碼使用 1000 個(gè)類作為輸出層。我只需要更改最后一部分嗎：# 3rd Fully Connected Layermodel.add(Dense(1000))model.add(Activation(‘relu’))# Add Dropoutmodel.add(Dropout(0.4))# Output Layermodel.add(Dense(17))model.add(Activation(‘softmax’))到：# 3rd Fully Connected Layermodel.add(Dense(11))model.add(Activation(‘relu’))# Add Dropoutmodel.add(Dropout(0.4))# Output Layermodel.add(Dense(17))model.add(Activation(‘softmax’))以便我可以測(cè)試我的數(shù)據(jù)集？PS 我是機(jī)器學(xué)習(xí)的新手，如果這是一個(gè)愚蠢的問(wèn)題，我很抱歉
1 回答
陪伴而非守候
TA貢獻(xiàn)1757條經(jīng)驗(yàn) 獲得超8個(gè)贊
如果您的數(shù)據(jù)集有 11 個(gè)類，請(qǐng)將最后一層的輸出數(shù)量更改為 11。試試這個(gè)：
# 3rd Fully Connected Layer
# This is only the output of a hidden layer, you don't have to change this
model.add(Dense(1000))
model.add(Activation(‘relu’))
# Add Dropout
model.add(Dropout(0.4))
# Output Layer
# This is what you want to change
model.add(Dense(11))
model.add(Activation(‘softmax’))
添加回答
舉報(bào)
0/150
提交
取消