From 37280c34a214b5e2601aaf0fce0f80052ebb185e Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <nikolai.hartmann@gmx.de>
Date: Thu, 29 Nov 2018 12:54:25 +0100
Subject: [PATCH] ensure fixed number of labels for adv target
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index 5ac5fff..c58f005 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -2060,7 +2060,7 @@ class ClassificationProjectDecorr(ClassificationProject):
bin_idx = np.digitize(out, binning)
# include overflow into last bin
bin_idx[bin_idx==len(binning)] = len(binning)-1
- out_list[i+1] = keras.utils.to_categorical(bin_idx)
+ out_list[i+1] = keras.utils.to_categorical(bin_idx, num_classes=len(binning))
return out_list
--
GitLab