使用XGBoost预测n日后的股价

Last Modified: Fri 17 Jun 2022 12:12:33 PM PST

介绍

使用XGBoost预测n日后的股价（使用tushare数据源）

软件架构

软件架构说明

安装教程

5月26日晚上传了新版本，程序能在更多的版本上跑了
由于要计算全部股票，tushare读取全部股票数据需要近1小时
计算结束后，在代码目录下生成XXXXXXresult.xls的文件，xxxxxx为当日日期
tushare目前开始提供因子数据，等tushare因子数据提供的足够多后可以，可以直接使用tushare的因子数据做特征工程
tushare的免费token可以使用。

配置tushare tokent：

# 复制示例代码 
cp example.env .env
# 改成自己的token
vim .env

注册免费tushare token https://tushare.pro/register?reg=286095
运行前将你的token更新到 GetData.py中ts.set_token('your token here')
XGBoost的调参一直做的不好，目前参数有些过拟合。后来明白了，是数据本身的问题。

使用说明

建议运行前创建一个单独python环境

# 创建名称为xpp的python环境
conda create -n xpp
conda activate xpp

git clone https://github.com/pchaos/XGBoostPredictPrice.git
cd XGBoostPredictPrice

python xgboostmodel.py
单元测试示例见tests目录;

# 测试例子
pytest tests/test_xgboostmodel.py

欢迎大家更新上传新特征，请使用tushare数据，最好注明需要的tushare点数