Как я могу создать собственный обратный вызов в Керасе? - PullRequest
0 голосов
/ 12 сентября 2018

Я заинтересован в создании обратного вызова при установке моей модели keras. Более подробно, я хотел бы получать сообщение от бот-телеграммы с val_acc каждый раз, когда эпоха заканчивается. Я знаю, что вы можете добавить callback_list в качестве параметра в classifier.fit (), но многие обратные вызовы предварительно созданы keras, и я не знаю, как добавить собственный.

Спасибо!

1 Ответ

0 голосов
/ 12 сентября 2018

Вот пример того, как я бы добавил точность проверки к обратному вызову:

class AccuracyHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        self.acc = []

    def on_epoch_end(self, batch, logs={}):
        self.acc.append(logs.get('val_acc'))

history = AccuracyHistory()

model.fit(x, y,
          ...
          callbacks=[history])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...