我试图从给定的图片中预测年龄。我在下面构建了模型,但问题是我在拟合模型时得到了非常大的loss值。accuracy
我认为问题在于选择了错误的损失函数(这里mean_squared_error)。这里有什么问题?
import tensorflow as tf
from tensorflow import keras
X = X.reshape(-1, image_size[0], image_size[1], 1)
model = keras.models.Sequential()
model.add(keras.layers.Conv2D(32, (5, 5), activation='relu', input_shape=X.shape[1:]))
model.add(keras.layers.MaxPooling2D((2, 2)))
model.add(keras.layers.Conv2D(32, (3, 3), activation='relu'))
model.add(keras.layers.MaxPooling2D(2, 2))
model.add(keras.layers.Conv2D(64, (3, 3), activation='relu'))
model.add(keras.layers.Flatten())
model.add(keras.layers.Dense(60, activation='relu'))
model.add(keras.layers.Dropout(0.4))
model.add(keras.layers.Dense(1, activation='softmax'))
model.compile(optimizer='adam', loss=keras.losses.mean_squared_error, metrics=['accuracy'])
model.fit(X, Y, epochs=170, shuffle=True, validation_split=0.1)
作为另一个问题,我的图层对于预测给定图片的数字是否正确?