Я пытаюсь использовать модель LSTM для данных временных рядов. Конкретным фоном данных, с которыми я работаю, является анализ настроений в Твиттере для прогнозирования будущих цен. Данные у меня выглядят так:
date mentions likes retweets polarity count Volume Close
2017-04-10 0.24 0.123 -0.58 0.211 0.58 0.98 0.87
2017-04-11 -0.56 0.532 0.77 0.231 -0.23 0.42 0.92
.
.
.
2019-01-10 0.23 0.356 -0.21 -0.682 0.23 -0.12 -0.23
Данные — это размер (608, 8), а функции, которые я планирую использовать, — это столбцы со 2 по 7, а цель, которую я предсказываю, — Close
(т. е. столбец 8). Я знаю, что модели LSTM требуют, чтобы ввод был в форме 3D-тензоров, поэтому я выполнил некоторые операции для преобразования и изменения формы данных:
x = np.asarray(data.iloc[:, 1:8])
y = np.asarray(data.iloc[:. 8])
x = x.reshape(x.shape[0], 1, x.shape[1])
После этого я попытался обучить модель LSTM как таковую:
batch_size = 200
model = Sequential()
model.add(LSTM(batch_size, input_dim=3, activation='relu', return_sequences=True))
model.add(Dropout(0.2))
model.add(LSTM(128, activation='relu'))
model.add(Dropout(0.1))
model.add(Dense(32, activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1))
model.compile(loss='mean_squared_error',
optimizer='rmsprop',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=15)
Запуск этой модели дает мне:
ValueError: Error when checking input: expected lstm_10_input to have
shape (None, 3) but got array with shape (1, 10)
Кто-нибудь знает, где я ошибся? Это связано с тем, как я подготовил данные, или я неправильно обучил модель?
Я читал много связанных вопросов в этом сообществе, а также статьи/блоги, но мне все еще не удается найти решение... Любая помощь приветствуется, спасибо!
Форма x должна быть формой (batch_size, timesteps, input_dim)
Первый аргумент LSTM — это не размер пакета, а выходной размер.
Пример:
df = pd.DataFrame(np.random.randn(100,9))
x_train = df.iloc[:,1:8].values
y_train = df.iloc[:,8].values
# No:of sample, times_steps, input_size (1 in your case)
x_train = x_train.reshape(x_train.shape[0],x_train.shape[1], 1)
model = Sequential()
# 16 outputs of first LSTM per time step
model.add(LSTM(16, input_dim=1, activation='relu', return_sequences=True))
model.add(Dropout(0.2))
model.add(LSTM(8, activation='relu'))
model.add(Dropout(0.1))
model.add(Dense(4, activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1))
model.compile(loss='mean_squared_error',
optimizer='rmsprop',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=15, batch_size=32)