bug fix attempt: large matrix conversion fails

4 years ago · f65b082fb1
parent bdd752550e
commit f65b082fb1
1 changed files with 2 additions and 1 deletions
--- 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