diff --git a/train.py b/train.py index 53824d0..8008310 100644 --- a/train.py +++ b/train.py @@ -72,7 +72,7 @@ def fit_lstm(train, batch_size2, nb_epoch, neurons): 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) - print(loss, accuracy) + #print(loss, accuracy) model.reset_states() return model