ierhon commited on
Commit
7bcdff2
1 Parent(s): 67f5bd6

Fix right padding error

Browse files
Files changed (1) hide show
  1. chatbot_constructor.py +1 -1
chatbot_constructor.py CHANGED
@@ -91,7 +91,7 @@ def train(message: str = "", regularization: float = 0.0001, dropout: float = 0.
91
  else:
92
  for key in dset:
93
  tokens = tokenizer.texts_to_sequences([key,])[0]
94
- X.append(np.array(list(tokens)+[0,]*inp_len[:inp_len]))
95
  y.append(dset[key])
96
 
97
  X = np.array(X)
 
91
  else:
92
  for key in dset:
93
  tokens = tokenizer.texts_to_sequences([key,])[0]
94
+ X.append(np.array((list(tokens)+[0,]*inp_len)[:inp_len]))
95
  y.append(dset[key])
96
 
97
  X = np.array(X)