如何在 keras 中的某个时代之后执行一些 python 代码

Question

我有简单的代码可以正常工作，但我需要创建大约 1% 的 ML 完成和 10% 的 ML 完成等......

   trainX = np.array(features_data)
    trainY = np.array(labels_data)
    model = Sequential()
    model.add(Dense(10, input_dim=input_dimensions, activation='relu'))
    model.add(Dense(1))
    model.compile(loss='mean_squared_error', optimizer='adam')
    model.fit(trainX, trainY, nb_epoch=3000, batch_size=2, verbose=2)
    model.save(model_location)

或者如何在纪元执行后执行一些 python 代码？

Epoch 1707/3000
0s - loss: 0.5908
Epoch 1708/3000
0s - loss: 0.4808
Epoch 1709/3000
// how to execute here some code on python
0s - loss: 0.7568
Epoch 1710/3000
0s - loss: 0.5906

Answer 1

您正在寻找Callback .它有一个特殊的方法 - on_epoch_end在每个纪元之后执行。文档可能会向您展示使用其API可以实现的不同方案。