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点数