bug fix: volume of data

5 years ago · e27624b697
parent 310d599d06
commit e27624b697
1 changed files with 1 additions and 1 deletions
--- a/data/gan.py
+++ b/data/gan.py
@ -593,7 +593,7 @@ class Predict(GNet):
                                #
                                # df =   pd.DataFrame(np.round(f)).astype(np.int32)
-                                df =   pd.DataFrame(np.round(f),dtype=int)
+                                df =   pd.DataFrame(np.round(f),dtype=np.uint8)
                                p = 0 not in df.sum(axis=1).values
                                x = df.sum(axis=1).values