bug fix , generator

5 years ago · c1a500fe4c
parent cd88a9660a
commit c1a500fe4c
1 changed files with 3 additions and 2 deletions
--- 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