【发布时间】:2017-08-21 05:54:29
【问题描述】:
我是 LSTM 的新手,我正在尝试训练一个模型来预测给定一年数据的 IP 的流量。数据集由 Kaggle https://www.kaggle.com/crawford/computer-network-traffic 提供。
这就是网络的建模方式
model = Sequential()
model.add(LSTM(128,input_shape=(trainX.shape[1], trainX.shape[2]),
activation='relu',return_sequences=True))
model.add(LSTM(32, return_sequences=True))
model.add(LSTM(10))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(trainX, trainY, epochs=10, batch_size=64, verbose=2)
你可以在我的内核https://www.kaggle.com/asindico/computer-network-traffic-eda/找到所有细节
这是我在 10 个 epoch 后得到的
蓝色表示实际值,红色表示预测值。
【问题讨论】:
-
这里有三件事: 1. 使用例如规范化你的目标。
StandardScaller来自sklearn。 2. 从 LSTM 的第一层删除relu,因为这实际上可能会损害训练。 3. 训练更长时间 -LSTM喜欢更长时间的训练。
标签: machine-learning deep-learning keras lstm tensor