‘EarlyStopping’ object has no attribute ‘_implements_train_batch_hooks’
EarlyStopping’ object has no attribute '_implements_train_batch_hooks’解决办法
在使用keras进行模型训练时遇到错误EarlyStopping’ object has no attribute ‘_implements_train_batch_hooks’,原始代码为:
from keras.callbacks import EarlyStopping
early_stopping = EarlyStopping(monitor=‘val_loss’, patience=8, verbose=2)
adam=Adam(lr=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-08, decay=0.0)
model2.compile(optimizer=adam,loss=keras.losses.mse, metrics=[‘mae’])
model2.fit(X_train, y_train,validation_data = (X_test, y_test), callbacks=[early_stopping],
verbose = 2,epochs=50, batch_size=512)
将from keras.callbacks import EarlyStopping改为:
from tensorflow.keras.callbacks import EarlyStopping
问题解决!