diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 32dd58b..38e0848 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -224,7 +224,7 @@ - + diff --git a/train.py b/train.py index 0eb0d5a..76ddb5e 100644 --- a/train.py +++ b/train.py @@ -62,6 +62,7 @@ def invert_scale(scaler, X, yhat): def fit_lstm(train, batch_size2, nb_epoch, neurons): X, y = train[:, 0:-1], train[:, -1] X = X.reshape(X.shape[0], 1, X.shape[1]) + print(X, y) model = Sequential() model.add(LSTM(neurons, batch_input_shape=(batch_size2, X.shape[1], X.shape[2]), stateful=True)) model.add(Dense(1))