Я пытаюсь использовать контрольные точки раннего останова и модели, чтобы сохранить лучшую модель при обучении нейронной сети с глубоким свертыванием. Однако я получаю следующую ошибку:
callback.set_model(model)
AttributeError: 'list' object has no attribute 'set_model'
Пока мой код:
model = Sequential()
###First block
model.add(Conv2D(100,kernel_size = (3,3),activation = 'relu',padding = 'same',input_shape=(12,11,1)))
model.add(Conv2D(100,kernel_size = (3,3),activation = 'relu',padding = 'same'))
model.add(MaxPooling2D(pool_size = (2,2)))
model.add(Dropout(0.20))
###Second block
model.add(Conv2D(128,kernel_size = (3,3),activation = 'relu',padding = 'same'))
model.add(Conv2D(128,kernel_size = (3,3),activation = 'relu',padding = 'same'))
model.add(MaxPooling2D(pool_size = (2,2)))
model.add(Dropout(0.10))
model.add(Flatten())
#model.add(Dense(100,activation = 'relu',kernel_regularizer=regularizers.l2(0.01)))
model.add(Dense(1000,activation = 'relu',kernel_regularizer=regularizers.l2(0.01)))
model.add(Dropout(0.30))
model.add(Dense(500,activation = 'relu',kernel_regularizer=regularizers.l2(0.01)))
model.add(Dropout(0.10))
#model.add(Dense(500,activation = 'relu',kernel_regularizer=regularizers.l2(0.01)))
#model.add(Dropout(0.15))
model.add(Dense(5,activation = 'softmax'))
model.compile(loss='categorical_crossentropy',
optimizer='adam',
metrics=['accuracy'])
earlystop = [EarlyStopping(monitor='val_acc', min_delta=0.001, patience=5,
verbose=1, mode='auto')]
outputModel = 'outputModel'
model_json = model.to_json()
with open(outputModel+".json", "w") as json_file:
json_file.write(model_json)
modWeightsFilepath=outputModel+"_weights.hdf5"
checkpoint = ModelCheckpoint(modWeightsFilepath, monitor='val_acc', verbose=1, save_best_only=True, save_weights_only=True, mode='auto')
callbacks_list = [earlystop,checkpoint]
history = model.fit(x_train, Y,
batch_size=100, ##number of observations per batch
epochs=100, ##Number of epochs
callbacks = callbacks_list,
verbose=1,
shuffle = True,
validation_split=0.2) ###Data for evaluation
Я не знаю, что я делаю не так. Я прочитал, что ModelCheckPoint и ранняя остановка должны быть заданы в виде списка, поэтому я явно сделаю это как:
callbacks_list = [earlystop,checkpoint]
Помощь будет оценена.