-
SARIMA模型:
import pandas as pd
import numpy as np
from statsmodels.tsa.statespace.sarimax import SARIMAX将日期转换为datetime格式
data['date'] = pd.to_datetime(data['date'], format='%Y%m%d')
创建SARIMA模型
model = SARIMAX(data['close'], order=(1, 0, 0), seasonal_order=(1, 1, 1, 12))
拟合模型
model_fit = model.fit()
预测未来3天的价格
future_dates = pd.date_range(start=data['date'].iloc[-1], periods=3, freq='D')
forecast = model_fit.get_forecast(steps=3)
predicted_values = forecast.predicted_mean输出预测结果
print(predicted_values)
-
简单移动平均线(SMA)模型:
计算简单移动平均线
data['SMA'] = data['close'].rolling(window=7).mean()
预测未来3天的价格(使用最近7天的平均值作为预测值)
last_7_days = data['close'].tail(7)
predicted_values = np.mean(last_7_days)输出预测结果
print(predicted_values)
-
指数加权移动平均线(EMA)模型:
计算指数加权移动平均线
data['EMA'] = data['close'].ewm(span=7, adjust=False).mean()
预测未来3天的价格(使用最近一个指数加权移动平均值作为预测值)
last_EMA = data['EMA'].iloc[-1]
predicted_values = last_EMA输出预测结果
print(predicted_values)
-
Bollinger带模型:
计算布林带指标
data['MA'] = data['close'].rolling(window=20).mean()
data['std'] = data['close'].rolling(window=20).std()
data['upper_band'] = data['MA'] + 2 * data['std']
data['lower_band'] = data['MA'] - 2 * data['std']预测未来3天的价格(使用最近一个布林带的上轨值作为预测值)
last_upper_band = data['upper_band'].iloc[-1]
predicted_values = last_upper_band输出预测结果
print(predicted_values)
-
相对强弱指标(RSI)模型:
计算相对强弱指标
data['delta'] = data['close'].diff()
data['gain'] = np.where(data['delta'] >= 0, data['delta'], 0)
data['loss'] = np.where(data['delta'] < 0, -data['delta'], 0)
data['avg_gain'] = data['gain'].rolling(window=14).mean()
data['avg_loss'] = data['loss'].rolling(window=14).mean()计算相对强弱指标
data['RS'] = data['avg_gain'] / data['avg_loss']
data['RSI'] = 100 - (100 / (1 + data['RS']))预测未来3天的价格(使用最近一个相对强弱指标值作为预测值)
last_RSI = data['RSI'].iloc[-1]
predicted_values = last_RSI输出预测结果
print(predicted_values)
-
随机指标(KD指标)模型:
计算随机指标(KD指标)
data['lowest_low'] = data['low'].rolling(window=9).min()
data['highest_high'] = data['high'].rolling(window=9).max()
data['%K'] = (data['close'] - data['lowest_low']) / (data['highest_high'] - data['lowest_low']) * 100
data['%D'] = data['%K'].rolling(window=3).mean()预测未来3天的价格(使用最近一个随机指标值作为预测值)
last_%K = data['%K'].iloc[-1]
predicted_values = last_%K输出预测结果
print(predicted_values)
-
线性回归模型:
from sklearn.linear_model import LinearRegression
创建线性回归模型
model = LinearRegression()
准备训练数据
X = data['date'].values.reshape(-1, 1)
y = data['close']拟合模型
model.fit(X, y)
预测未来3天的价格
future_dates = pd.date_range(start=data['date'].iloc[-1], periods=3, freq='D')
X_future = future_dates.values.reshape(-1, 1)
predicted_values = model.predict(X_future)输出预测结果
print(predicted_values)
-
随机森林回归模型:
from sklearn.ensemble import RandomForestRegressor
创建随机森林回归模型
model = RandomForestRegressor()
准备训练数据
X = data['date'].values.reshape(-1, 1)
y = data['close']拟合模型
model.fit(X, y)
预测未来3天的价格
future_dates = pd.date_range(start=data['date'].iloc[-1], periods=3, freq='D')
X_future = future_dates.values.reshape(-1, 1)
predicted_values = model.predict(X_future)输出预测结果
print(predicted_values)
-
支持向量回归(SVR)模型:
from sklearn.svm import SVR
from sklearn.preprocessing import StandardScaler创建支持向量回归模型
model = SVR()
准备训练数据
X = data['date'].values.reshape(-1, 1)
y = data['close']特征缩放
scaler = StandardScaler()
X_scaled = scaler.fit_transform(X)拟合模型
model.fit(X_scaled, y)
预测未来3天的价格
future_dates = pd.date_range(start=data['date'].iloc[-1], periods=3, freq='D')
X_future = future_dates.values.reshape(-1, 1)
X_future_scaled = scaler.transform(X_future)
predicted_values = model.predict(X_future_scaled)输出预测结果
print(predicted_values)
【@胡锡进】大模型量化分析- 福耀玻璃 600660.SH
civilpy2023-10-07 17:50
相关推荐
今天也要加油丫8 分钟前
`re.compile(r“(<.*?>)“)` 如何有效地从给定字符串中提取出所有符合 `<...>` 格式的引用农民小飞侠1 小时前
python AutoGen接入开源模型xLAM-7b-fc-r,测试function calling的功能战神刘玉栋1 小时前
《程序猿之设计模式实战 · 观察者模式》敲代码不忘补水1 小时前
Python 项目实践:简单的计算器鸽芷咕2 小时前
【Python报错已解决】ModuleNotFoundError: No module named ‘paddle‘子午3 小时前
动物识别系统Python+卷积神经网络算法+TensorFlow+人工智能+图像识别+计算机毕业设计项目风等雨归期3 小时前
【python】【绘制小程序】动态爱心绘制Adolf_19933 小时前
Flask-JWT-Extended登录验证, 不用自定义冯宝宝^3 小时前
基于mongodb+flask(Python)+vue的实验室器材管理系统叫我:松哥3 小时前
基于Python flask的医院管理学院,医生能够增加/删除/修改/删除病人的数据信息,有可视化分析