From c1a500fe4c3d18fa0606b2e68ad980515cd30f52 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Thu, 20 Feb 2020 23:08:35 -0600 Subject: [PATCH] bug fix , generator --- data/gan.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/data/gan.py b/data/gan.py index 2083f69..4c05566 100644 --- a/data/gan.py +++ b/data/gan.py @@ -568,8 +568,9 @@ class Predict(GNet): # df = ( pd.DataFrame(np.round(f).astype(np.int32))) p = 0 not in df.sum(axis=1).values - - if p: + x = df.sum(axis=1).values + print ( [np.sum(x),x.size]) + if np.divide( np.sum(x), x.size) : found.append(df) if len(found) == NTH_VALID_CANDIDATE or i == CANDIDATE_COUNT: break