我构建了一个脚本,将我的最佳模型保存到文件系统中,并且我希望它还将有关模型预期性能的数据记录到不同的文本文件中。
当前代码:
model = Sequential()
model.add(Dense(64, input_dim=14, init='uniform'))
model.add(LeakyReLU(alpha=0.3))
model.add(BatchNormalization(epsilon=1e-06, mode=0, momentum=0.9, weights=None))
model.add(Dropout(0.5))
model.add(Dense(64, init='uniform'))
model.add(LeakyReLU(alpha=0.3))
model.add(BatchNormalization(epsilon=1e-06, mode=0, momentum=0.9, weights=None))
model.add(Dropout(0.5))
model.add(Dense(2, init='uniform'))
model.add(Activation('softmax'))
sgd = SGD(lr=0.1, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='binary_crossentropy', optimizer=sgd)
checkpointer = ModelCheckpoint(filepath="/Desktop/SquareSpace/checkpointer", verbose=2, save_best_only=True)
model.fit(X_train, y_train, nb_epoch=100, batch_size=50, show_accuracy=True, validation_split=0.2, verbose = 2, callbacks=[checkpointer])
我目前将我最好的模型保存为 HDF5 文件。有没有办法使用 Keras 在不同的文本文件中记录保存在 HDF5 文件中的最佳模型的性能?
编辑:有没有办法在这里使用它http://keras.io/callbacks/?