小编dJO*_*MMY的帖子

在keras中保存最佳模型

我在keras中训练模型时使用以下代码

from keras.callbacks import EarlyStopping

model = Sequential()
model.add(Dense(100, activation='relu', input_shape = input_shape))
model.add(Dense(1))

model_2.compile(optimizer='adam', loss='mean_squared_error', metrics=['accuracy'])


model.fit(X, y, epochs=15, validation_split=0.4, callbacks=[early_stopping_monitor], verbose=False)

model.predict(X_test)
Run Code Online (Sandbox Code Playgroud)

但是最近我想得到保存最好的训练模型,因为我训练的数据在"高val_loss vs epochs"图中给出了很多高峰,我想从模型中使用最好的一个.

是否有任何方法或功能可以帮助解决这个问题?

neural-network deep-learning keras

19
推荐指数
3
解决办法
2万
查看次数

标签 统计

deep-learning ×1

keras ×1

neural-network ×1