【发布时间】:2022-01-12 12:43:09
【问题描述】:
我将 GridSearchCV 与 keras 一起使用,我想绘制和分析训练与验证历史。但是,我已经检查了文档并确实搜索了 SO,但是当使用 GridSearchCV 拟合模型时,我找不到获取验证历史记录(即每个时期的分数)的方法。我能够在回调中获取培训历史记录,但不是验证记录。问题是一些模型过拟合很多,我希望能够看到调整参数如何影响过拟合。
我正在像这样使用 GridSearchCV:
class MyCallback(keras_callbacks.Callback):
def on_train_end(self, logs=None):
# here I can get the model history from self.model.history.history
def create_model(...):
...
model = Model(...)
model.compile(optimizer=optimizer, loss="binary_crossentropy", metrics=['acc'])
return model
callbacks = [MyCallback()]
model = KerasClassifier(build_fn=create_model, verbose=3)
grid = GridSearchCV(estimator=model, param_grid=param_grid, n_jobs=4, cv=3, verbose=10, return_train_score=True)
grid_result = grid.fit(X_train["padded"], y_train["binary"], epochs=30, batch_size=16, callbacks=callbacks)
【问题讨论】:
标签: python tensorflow keras scikit-learn