diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 196d043..64db22e 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -227,7 +227,7 @@ - + diff --git a/train.py b/train.py index 7437a49..53824d0 100644 --- a/train.py +++ b/train.py @@ -71,7 +71,7 @@ def fit_lstm(train, batch_size2, nb_epoch, neurons): for i in range(nb_epoch): print("Epoch {}/{}".format(i, nb_epoch)) model.fit(X, y, epochs=1, batch_size=batch_size2, verbose=0, shuffle=False) - loss, accuracy = model.evaluate(X, y) + #loss, accuracy = model.evaluate(X, y) print(loss, accuracy) model.reset_states() return model