python预测股价代码,使用SVM
时间: 2023-10-24 21:24:16 浏览: 97
一个python股价预测代码
以下是一个基于 SVM 的股价预测代码示例:
```python
import pandas as pd
from sklearn.svm import SVR
import matplotlib.pyplot as plt
# 读取数据
df = pd.read_csv('stock_prices.csv')
# 将日期转换为 datetime 类型
df['Date'] = pd.to_datetime(df['Date'])
# 选取需要预测的股票价格列
data = df[['Date', 'Close']]
# 将日期作为索引
data.set_index('Date', inplace=True)
# 分离训练集和测试集
train_data = data[:'2019']
test_data = data['2020':]
# 创建模型
model = SVR(kernel='linear', C=1e3)
# 训练模型
model.fit(train_data.index.values.reshape(-1, 1), train_data['Close'])
# 进行预测
predictions = model.predict(test_data.index.values.reshape(-1, 1))
# 绘制预测结果和实际结果
plt.plot(test_data.index, predictions, label='Predicted')
plt.plot(test_data.index, test_data['Close'], label='Actual')
plt.legend()
plt.show()
```
在这个示例中,我们使用了一个线性的 SVM 模型,使用前 80% 的数据来训练模型,然后使用后 20% 的数据来测试模型。我们使用预测结果和实际结果来绘制一个简单的图表,以便我们可以比较它们。请注意,这只是一个示例,您需要根据您的数据和需求进行调整。
阅读全文