Spaces:
Sleeping
Sleeping
Fix epochs not changing mistake
Browse files- chatbot_constructor.py +1 -1
chatbot_constructor.py
CHANGED
|
@@ -99,7 +99,7 @@ def train(message: str = "", regularization: float = 0.0001, dropout: float = 0.
|
|
| 99 |
|
| 100 |
model.compile(optimizer=RMSprop(learning_rate=learning_rate), loss="sparse_categorical_crossentropy", metrics=["accuracy",])
|
| 101 |
|
| 102 |
-
model.fit(X, y, epochs=
|
| 103 |
model.save(f"cache/{data_hash}")
|
| 104 |
tokens = tokenizer.texts_to_sequences([message,])[0]
|
| 105 |
prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
|
|
|
|
| 99 |
|
| 100 |
model.compile(optimizer=RMSprop(learning_rate=learning_rate), loss="sparse_categorical_crossentropy", metrics=["accuracy",])
|
| 101 |
|
| 102 |
+
model.fit(X, y, epochs=epochs, batch_size=8, workers=4, use_multiprocessing=True)
|
| 103 |
model.save(f"cache/{data_hash}")
|
| 104 |
tokens = tokenizer.texts_to_sequences([message,])[0]
|
| 105 |
prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
|