Я получаю сообщение об ошибке: объект «ModifiedTensorBoard» не имеет атрибута «_write_logs» при выполнении этого кода для обучения Deep Q Network - PullRequest
Новая
       35

Я получаю сообщение об ошибке: объект «ModifiedTensorBoard» не имеет атрибута «_write_logs» при выполнении этого кода для обучения Deep Q Network

0 голосов
/ 01 мая 2020

Я использую следующий код: 

class ModifiedTensorBoard(TensorBoard):

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.step = 1
        self.writer = tf.summary.create_file_writer(self.log_dir)

    def set_model(self, model):
        pass

    def on_epoch_end(self, epoch, logs=None):
        self.update_stats(**logs)

    def on_batch_end(self, batch, logs=None):
        pass

    def on_train_end(self, _):
        pass

    def update_stats(self, **stats):
        self._write_logs(stats, self.step)

Получена ошибка: 

    232     def update_stats(self, **stats):
--> 233         self._write_logs(stats, self.step)
    234 
    235 

AttributeError: 'ModifiedTensorBoard' object has no attribute '_write_logs'

Может кто-нибудь подсказать, как решить эту проблему? Спасибо.

...