Skip to content
Snippets Groups Projects
Commit 4658a837 authored by Dennis Noll's avatar Dennis Noll
Browse files

[keras] PlotMulticlass kwarg

parent efb7e16f
No related branches found
No related tags found
No related merge requests found
......@@ -483,7 +483,7 @@ class PlotMulticlass(TFSummaryCallback):
plt.close("all")
gc.collect()
def make_eval_plots(self, epoch):
def make_eval_plots(self, epoch=0):
prediction = self.model.predict(self.x, batch_size=4096)
truth = self.truth
imgs = {}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment