keras构建LSTM模型，预测带高度的经纬度位置

原始时间序列数据如下所示，我们将纬度、经度、高度作为输入

其轨迹经纬度如下图所示：

用keras创建一个三层的LSTM网络训练模型如下图所示：

import numpy as np
from keras.layers.core import Dense, Activation, Dropout
from keras.layers import LSTM
from keras.models import Sequential, load_model
from keras.callbacks import Callback
import keras.backend.tensorflow_backend as KTF
import tensorflow as tf
import pandas as pd
import os
import  keras.callbacks
import matplotlib.pyplot as plt

#设定为自增长
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
KTF.set_session(session)


def trainModel(train_X, train_Y):
    '''
    trainX，trainY: 训练LSTM模型所需要的数据
    '''
    model = Sequential()  # 定义一个堆叠的顺序模型
    model.add(LSTM(
        240,
        input_shape=(train_X.shape[1], train_X.shape[2]),
        return_sequen