Spaces:
Sleeping
Sleeping
Fix RMSprop name error
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_le
|
|
62 |
X = np.array(X)
|
63 |
y = np.array(y)
|
64 |
|
65 |
-
model.compile(optimizer=
|
66 |
|
67 |
model.fit(X, y, epochs=16, batch_size=8, workers=4, use_multiprocessing=True)
|
68 |
model.save(f"cache/{data_hash}")
|
|
|
62 |
X = np.array(X)
|
63 |
y = np.array(y)
|
64 |
|
65 |
+
model.compile(optimizer=RMSprop(learning_rate=learning_rate), loss="sparse_categorical_crossentropy", metrics=["accuracy",])
|
66 |
|
67 |
model.fit(X, y, epochs=16, batch_size=8, workers=4, use_multiprocessing=True)
|
68 |
model.save(f"cache/{data_hash}")
|