From f65b082fb129cbb37091a35d8823218c766882e7 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Tue, 30 Mar 2021 17:01:59 -0500 Subject: [PATCH] bug fix attempt: large matrix conversion fails --- data/gan.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/data/gan.py b/data/gan.py index c61d1b1..767a24b 100644 --- a/data/gan.py +++ b/data/gan.py @@ -603,7 +603,8 @@ class Predict(GNet): # # df = pd.DataFrame(np.round(f)).astype(np.int32) - candidates.append (np.round(_matrix).astype(np.int64)) + # candidates.append (np.round(_matrix).astype(np.int64)) + candidates.append( [np.round(row).astype(int) for row in _matrix]) # return candidates[0] if len(candidates) == 1 else candidates return candidates