ierhon commited on
Commit
ce0df62
·
verified ·
1 Parent(s): f72cb30

Add longer training

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def train(data: str, message: str):
68
 
69
  model.compile(loss="categorical_crossentropy", metrics=["accuracy",])
70
 
71
- model.fit(X, y, epochs=10, batch_size=8, workers=4, use_multiprocessing=True)
72
  model.save("{data_hash}.keras")
73
  tokens = tokenizer.texts_to_sequences([message,])[0]
74
  prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
 
68
 
69
  model.compile(loss="categorical_crossentropy", metrics=["accuracy",])
70
 
71
+ model.fit(X, y, epochs=64, batch_size=8, workers=4, use_multiprocessing=True)
72
  model.save("{data_hash}.keras")
73
  tokens = tokenizer.texts_to_sequences([message,])[0]
74
  prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]