committed Andy Mattausch's code under related/EP/
This commit is contained in:
11
related/EP/src/LossHistory.py
Normal file
11
related/EP/src/LossHistory.py
Normal 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)
|
||||
Reference in New Issue
Block a user