committed Andy Mattausch's code under related/EP/

This commit is contained in:
Thomas Gabor
2019-03-02 18:51:19 +01:00
parent 0105eb6998
commit 3a170d886b
11 changed files with 3984 additions and 0 deletions

View File

@@ -0,0 +1,11 @@
import keras
class LossHistory(keras.callbacks.Callback):
def on_train_begin(self, logs={}):
self.losses = []
def on_batch_end(self, batch, logs={}):
self.losses.append(logs.get('loss'))
def addLoss(self,loss):
self.losses.append(loss)