簡體   English   中英

在keras中的每個紀元后如何查看驗證錯誤

[英]How to see the validation error after each epoch in 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)

問題在於它過度擬合。 在每個時期之后如何查看驗證錯誤?

您可以將參數傳輸到KerasRegressor fit方法:

validation_split:浮點型(0. <x <1)。 用作保留驗證數據的數據分數。 validation_data:用作保留驗證數據的元組(x_val,y_val)或元組(x_val,y_val,val_sample_weights)。 將覆蓋validation_split。

通過管道擬合方法

** fit_params:字符串->對象的字典傳遞給每個步驟的fit方法的參數,其中每個參數名稱都帶有前綴,以便步驟s的參數p具有鍵s__p。

例:

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

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM