根据表格数据,利用线性回归预测股票涨跌写一个机器学习运行代码
时间: 2024-10-21 11:14:04 浏览: 24
Python数据挖掘与机器学习开发实战_基于线性回归的股票预测_优秀案例实例源代码源码.zip
5星 · 资源好评率100%
在Python中,我们可以使用`pandas`, `numpy`, 和 `sklearn`库来根据表格数据(通常包含历史股价和可能影响股价的特征)执行线性回归模型预测股票涨跌。以下是一个简单的示例代码:
```python
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn import metrics
# 假设df是你的DataFrame,包含日期('Date'),开盘价('Open'), 收盘价('Close'), 其他特征列如'MovingAverage', 'Volume'等
data = pd.read_csv('stock_data.csv') # 读取CSV文件
# 数据预处理:选择特征和目标变量
X = data[['MovingAverage', 'Volume']] # 特征列
y = data['Close'] # 目标变量 - 收盘价
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建并训练线性回归模型
model = LinearRegression()
model.fit(X_train, y_train)
# 预测
y_pred = model.predict(X_test)
# 评估模型性能
print(f"Mean Absolute Error (MAE): {metrics.mean_absolute_error(y_test, y_pred)}")
print(f"Mean Squared Error (MSE): {metrics.mean_squared_error(y_test, y_pred)}")
print(f"Root Mean Squared Error (RMSE): {np.sqrt(metrics.mean_squared_error(y_test, y_pred))}")
# 如果你想基于新的输入数据预测涨跌,你可以这样做(假设new_data是新数据)
# new_data = ... # 新的数据特征
# prediction = model.predict(new_data)
```
注意:股票价格受许多因素影响,包括市场情绪、宏观经济状况、政策变动等,线性回归可能不是最准确的预测工具。此外,实际应用中还会涉及到时间序列分析、技术指标和更复杂的模型。
阅读全文