From 78718b6c42793d8df227f4d08f800f5dc4b89cfb Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Fri, 6 Mar 2020 15:39:13 -0600 Subject: [PATCH] ... --- data/gan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/gan.py b/data/gan.py index 80c3f8e..7952f23 100644 --- a/data/gan.py +++ b/data/gan.py @@ -584,7 +584,7 @@ class Predict(GNet): p = 0 not in df.sum(axis=1).values x = df.sum(axis=1).values - if x.max() == 1 and np.divide( np.sum(x), x.size) > .9 or p and np.sum(x) == x.size and x.size == self.values.size: + if np.divide( np.sum(x), x.size) > .9 or p and np.sum(x) == x.size and x.size == self.values.size: ratio.append(np.divide( np.sum(x), x.size)) found.append(df) if i == CANDIDATE_COUNT: