จะให้ฉันเพิ่มเติมลดการสูญเสียค่าในซีเอ็นเอ็นเลยรุ่น? [ปิด]

0

คำถาม

ฉันพยายามจะสร้างซีเอ็นเอ็นเลยเพื่อจำแนก fruits. ฉันรับประสบการณ์สูงสูญเสียค่าและฉันกำลังพยายามเพื่อคุมห้องระบบประสาทพวกมันมากเท่าที่ฉันสามารถแต่ฉันไม่แน่ใจว่าวิธีที่จะปรับปรุงของฉันนางแบบมากไปกว่านี้

นี่คือรหัส:

model96 = tf.keras.Sequential()

#Architecture
model96.add(tf.keras.layers.Conv2D(filters = 32,
                                 kernel_size = (3, 3),
                                 activation = "relu",
                                 input_shape = (96, 96, 3)))

model96.add(tf.keras.layers.Conv2D(filters = 32,
                                 kernel_size = (3, 3),
                                 activation = "relu"))

model96.add(tf.keras.layers.MaxPooling2D(pool_size=(2, 2)))

model96.add(tf.keras.layers.Dropout(rate=0.25))

model96.add(tf.keras.layers.Flatten())

model96.add(tf.keras.layers.Dense(units=128, activation='relu'))

model96.add(tf.keras.layers.Dropout(rate=0.5))

#output layer
model96.add(tf.keras.layers.Dense(units=4, activation='softmax'))

#Loss function
model96.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])

#Train model
hist96 = model96.fit(x=x_train96_norm, y=y_train, epochs=100)

#Test and Evaluate
print("Performance with test data:")
loss96, accuracy96 = model96.evaluate(x=x_test96_norm, y=y_test)
print('loss =', loss96)
print('accuracy =', accuracy96)

ระหว่างฝึกสุดท้ายสูญเสียค่าเป็น 0.0153 และสุดท้ายความถูกต้องค่าเป็น 0.9958 อย่างไรก็ตามระหว่างการทดสอบรุ่นมีคะแนน: loss = 1.0462701320648193 แล้ว accuracy = 0.8666666746139526

1

คำตอบที่ดีที่สุด

2

ปัญหาของคุณดูเหมือนจะเป็นคลาสสิค overfitting ปัญหาแล้ว คุณสามารถเพิ่ม EarlyStopping จะหลีกเลี่ยงเรื่องนี้ EarlyStopping จะหยุดรับการฝึกไม่นับโพรเซสทันทีที่การตรวจสอบการสูญเสียหยุด decreasing. รหัสมันค่อนข่างตรงไปตรงมา

callback = tf.keras.callbacks.EarlyStopping(monitor='loss', patience=3)

hist96 = model96.fit(x=x_train96_norm, y=y_train, epochs=100, callbacks=[callback])

2021-11-24 07:36:48

ในภาษาอื่นๆ

หน้านี้อยู่ในภาษาอื่นๆ

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................

ดังอยู่ในนี้หมวดหมู่

ดังคำถามอยู่ในนี้หมวดหมู่