diff --git a/data/bridge.py b/data/bridge.py index ac79272..a86deef 100644 --- a/data/bridge.py +++ b/data/bridge.py @@ -174,6 +174,7 @@ class Binary : i = np.random.choice(col_count,size) values = values[-i] col_count = N + matrix = [ np.zeros(col_count,dtype=np.float32) for i in np.arange(row_count)] @@ -203,10 +204,12 @@ class Binary : # Let's treat the case of missing values i.e nulls # row_count,col_count = column.size,values.size - if row_count * col_count > size and row_count < size: - N = np.divide(size,row_count).astype(int) - i = np.random.choice(col_count,N) + if col_count > size : + # N = np.divide(size,row_count).astype(int) + # N = + i = np.random.choice(col_count,size) values = values[-i] + col_count = N return values def _Export(self,df) :