如何在keras中的每个纪元后看到验证错误

时间:2017-10-13 12:28:52

标签: python machine-learning keras

我正在使用keras训练模型进行回归。我的代码如下:

estimators = []
estimators.append(('standardize', StandardScaler()))
estimators.append(('mlp', KerasRegressor(build_fn=baseline_model, epochs=100, batch_size=32, verbose=2)))
pipeline = Pipeline(estimators)
X_train, X_test, y_train, y_test = train_test_split(X, Y,
                                                    train_size=0.75, test_size=0.25)
pipeline.fit(X_train, y_train)
  

问题是它显然过度拟合。我怎么能看到   每个纪元后的验证错误?

1 个答案:

答案 0 :(得分:2)

您可以将parameters传送给KerasRegressor适合方法:

  

validation_split:float(0< x< 1)。要用作数据的分数   保持验证数据。 validation_data:tuple(x_val,y_val)或   元组(x_val,y_val,val_sample_weights)用作保持   验证数据。将覆盖validation_split。

通过管道fit method

  

** fit_params:字符串的字典 - > object传递给每个步骤的fit方法的参数,其中每个参数名称都以此为前缀   步骤s的参数p有键s__p。

示例:

pipeline.fit(X_train, y_train, mlp__validation_split=0.3)