以下是一个基于LSTM的船舶轨迹预测代码示例,可以根据自己的数据进行修改和调整:
```python
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from keras.models import Sequential
from keras.layers import LSTM, Dense
# 加载数据
data = pd.read_csv('ship_trajectory.csv')
# 数据预处理
data = data.iloc[:, 1:]
data = data.drop(columns=['MMSI', 'BaseDateTime', 'LAT', 'LON', 'SOG', 'COG'])
data = data.dropna()
data = data.astype('float32')
# 划分训练集和测试集
train_size = int(len(data) * 0.8)
test_size = len(data) - train_size
train_data, test_data = data.iloc[0:train_size,:], data.iloc[train_size:len(data),:]
# 将数据转换为适合LSTM的格式
def create_dataset(dataset, look_back=1):
X, Y = [], []
for i in range(len(dataset)-look_back-1):
a = dataset[i:(i+look_back), 0]
X.append(a)
Y.append(dataset[i + look_back, 0])
return np.array(X), np.array(Y)
look_back = 30
trainX, trainY = create_dataset(train_data.values, look_back)
testX, testY = create_dataset(test_data.values, look_back)
# 调整输入数据的形状
trainX = np.reshape(trainX, (trainX.shape[0], trainX.shape[1], 1))
testX = np.reshape(testX, (testX.shape[0], testX.shape[1], 1))
# 构建LSTM模型
model = Sequential()
model.add(LSTM(64, input_shape=(look_back, 1)))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
# 训练模型
model.fit(trainX, trainY, epochs=100, batch_size=64)
# 预测结果
train_predict = model.predict(trainX)
test_predict = model.predict(testX)
# 绘制预测结果和实际结果的对比图
train_predict_plot = np.empty_like(data)
train_predict_plot[:, :] = np.nan
train_predict_plot[look_back:len(train_predict)+look_back, :] = train_predict
test_predict_plot = np.empty_like(data)
test_predict_plot[:, :] = np.nan
test_predict_plot[len(train_predict)+(look_back*2)+1:len(data)-1, :] = test_predict
plt.plot(data)
plt.plot(train_predict_plot)
plt.plot(test_predict_plot)
plt.show()
```
需要注意的是,这里输入的数据必须是一个二维数组,第一维是时间序列,第二维是要预测的变量。在这个例子中,我们只预测了一种变量,即船舶速度。如果要预测多个变量,需要将它们按照某种规律组合成一个二维数组。此外,在预测结果和实际结果的对比图上,绿色线表示训练集预测结果,红色线表示测试集预测结果,蓝色线表示实际结果。