560cb94519ad5a5eed3adc323a9e1ad3cf0690e9,keras/callbacks.py,History,on_batch_end,#History#Any#Any#,165

Before Change

    def on_batch_end(self, batch, logs={}):
        batch_size = logs.get("size", 0)
        self.seen += batch_size
        self.tot_loss += logs.get("loss", 0.) * batch_size
        if self.params["show_accuracy"]:
            self.tot_accuracy += logs.get("accuracy", 0.) * batch_size

    def on_epoch_end(self, epoch, logs={}):
        val_loss = logs.get("val_loss")
        val_acc = logs.get("val_accuracy")
        self.epoch.append(epoch)

After Change

        self.totals = {}

    def on_batch_end(self, batch, logs={}):
        batch_size = logs.get("size", 0)
        self.seen += batch_size
        for k, v in logs.items():
            if k in self.totals:
                self.totals[k] += v * batch_size
            else:
                self.totals[k] = v * batch_size

    def on_epoch_end(self, epoch, logs={}):
        self.epoch.append(epoch)
        for k, v in self.totals.items():
            self.history[k] = v / self.seen

Instances