### 完整的 Python 实现股票价格预测模型 #### LSTM 模型用于股票价格预测 为了构建一个完整的 LSTM 模型来预测股票价格，以下是详细的代码实现： ```python import numpy as np import pandas as pd from sklearn.preprocessing import MinMaxScaler from keras.models import Sequential from keras.layers import Dense, LSTM, Dropout # 加载数据并预处理 data = pd.read_csv('stock_data.csv') # 替换为实际的数据文件路径 training_set = data.iloc[:, 1:2].values # 获取收盘价列作为训练集 scaler = MinMaxScaler(feature_range=(0, 1)) scaled_training_set = scaler.fit_transform(training_set) X_train = [] y_train = [] for i in range(60, len(scaled_training_set)): X_train.append(scaled_training_set[i-60:i, 0]) y_train.append(scaled_training_set[i, 0]) X_train, y_train = np.array(X_train), np.array(y_train) X_train = np.reshape(X_train, (X_train.shape[0], X_train.shape[1], 1)) # 构建LSTM模型 model = Sequential() model.add(LSTM(units=50, return_sequences=True, input_shape=(X_train.shape[1], 1))) model.add(Dropout(0.2)) model.add(LSTM(units=50, return_sequences=False)) model.add(Dropout(0.2)) model.add(Dense(units=1)) model.compile(optimizer='adam', loss='mean_squared_error') model.fit(X_train, y_train, epochs=100, batch_size=32) # 测试数据准备 test_data = pd.read_csv('test_stock_data.csv') # 替换为实际的测试数据文件路径 real_stock_price = test_data.iloc[:, 1:2].values dataset_total = pd.concat((data['Close'], test_data['Close']), axis=0) inputs = dataset_total[len(dataset_total) - len(test_data) - 60:].values inputs = inputs.reshape(-1, 1) inputs = scaler.transform(inputs) X_test = [] for i in range(60, len(inputs)): X_test.append(inputs[i-60:i, 0]) X_test = np.array(X_test) X_test = np.reshape(X_test, (X_test.shape[0], X_test.shape[1], 1)) predicted_prices = model.predict(X_test) # 使用模型进行预测[^1] predicted_stock_price = scaler.inverse_transform(predicted_prices) print("Predicted Prices:", predicted_stock_price) ``` 这段代码展示了如何加载、预处理股票市场数据，并利用Keras库中的LSTM层建立神经网络来进行股价走势预测。 #### ARIMA 模型用于股票价格预测 对于ARIMA模型，则可以按照如下方式编写代码： ```python import statsmodels.api as sm import pandas as pd # 数据读取与差分操作 data = pd.read_csv('stock_data.csv')['Close'] # 只保留收盘价一列 diff_num = 1 # 差分阶数可以根据ADF检验调整 differenced_series = data.diff(periods=diff_num).dropna() # 参数估计 p = 5 # 自回归项数目 q = 2 # 移动平均项数目 forecast_num = 10 # 预测未来期数 arima_model = sm.tsa.ARIMA(differenced_series, order=(p, diff_num, q)).fit() print('模型报告为：\n', arima_model.summary()) print("预测结果：") print(arima_model.forecast(steps=forecast_num)[0]) forecast_result = arima_model.get_forecast(steps=forecast_num) table = pd.DataFrame(forecast_result.conf_int(alpha=0.05)) print("预测区间:\n", table) # 输出置信区间的上下限[^2] ``` 此部分实现了基于`statsmodels`包的时间序列分析工具——ARIMA模型的应用实例。通过设定合适的参数组合以及对原始时间序列做适当次数的一阶差分转换成平稳过程之后再拟合ARIMA模型完成对未来一段时间内的股市行情变化趋势做出合理推测的任务目标。