From 8b5ab23e6506f0cfce5bc7845355af4a56003d72 Mon Sep 17 00:00:00 2001
From: Dennis Noll <github_001@nollde.de>
Date: Wed, 14 Oct 2020 17:00:15 +0200
Subject: [PATCH] [keras] fixes bug in plotting callback
---
keras.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/keras.py b/keras.py
index 5248cbe..6c5d7bb 100644
--- a/keras.py
+++ b/keras.py
@@ -410,8 +410,8 @@ class PlotMulticlass(TFSummaryCallback):
):
super().__init__(**kwargs)
self.x = x
- self.truth = y
- self.sample_weight = sample_weight
+ self.truth = y[0]
+ self.sample_weight = sample_weight[0]
self.class_names = class_names
self.plot_inputs = plot_inputs
self.columns = columns
--
GitLab