willspag / seqgan Goto Github PK

The following code :
smiles = ['G' for _ in range(gen_batch_size)]
gen_samples = np.array(self.st.one_hot_encode(smiles))
gen_samples = np.reshape(gen_samples,(gen_batch_size,1))

I set gen_batch_size to 128 and when I ran the second line of code I got an array of size (1,128,47) ,but np.reshape can't set size to (128,1) .This is wrong , I was wondering if there is another way or how much gen_batch_size should be set to work properly.