11 lines
284 B
Python
11 lines
284 B
Python
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) |