ierhon commited on
Commit
228e5ed
1 Parent(s): f5ee27b

Fix RMSprop name error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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}")
 
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}")