Welcome toVigges Developer Community-Open, Learning,Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
1.4k views
in Technique[技术] by (71.8m points)

tensorflow - Validation accuracy not changing while loss is decreasing in keras image classification?


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)

in model.compile your loss is incorrect. It should be

loss='BinaryCrossentropy'

In your model the last layer should be

tf.keras.layers.Dense(1,activation='sigmoid')

Alternatively you can keep everything as is but change class_mode in the train_data_gen and val_data_gen to

class_mode='sparse' 

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to Vigges Developer Community for programmer and developer-Open, Learning and Share
...