# 加載數據
series = read_csv('data_set/shampoo-sales.csv', header=0, parse_dates=[0],
index_col=0, squeeze=True, date_parser=parser)
# 配置網絡信息
n_lag = 1
n_seq = 3
n_test = 10
n_epochs = 1500
n_batch = 1
n_neurons = 1
# 準備數據
scaler, train, test = prepare_data(series, n_test, n_lag, n_seq)
# 準備預測模型
model = fit_lstm(train, n_lag, n_seq, n_batch, n_epochs, n_neurons)