def __init__(self, generator, steps):
History.__init__(self)
self.generator = generator
self.steps = steps
self.val_label = None
g1, g2 = itertools.tee(self.generator, 2)
self.generator = g1
for i in range(self.steps):
label_y = g2.__next__()[1]
if self.val_label is None:
self.val_label = np.array(label_y)
else:
self.val_label = np.concatenate(
(self.val_label, np.array(label_y)))
def __init__(self, file_name='history.png'):
History.__init__(self)
self.file_name = file_name
self.losses = []
self.f = plt.figure(0)
self.ax = self.f.gca()
