FinRL

首个开源的金融强化学习框架，让 AI 帮你交易。

金融数据集成

内置了 Yahoo Finance, Google Finance 等多个数据源，一键获取股票、加密货币历史数据。

SOTA 算法

针对金融场景优化了 DRL 算法，并提供了多种 Baseline 策略进行对比。

全流程支持

从数据处理、特征工程、模型训练到回测（Backtest），提供了一站式解决方案。

安装与使用

install.sh
pip install finrl

数据下载与环境构建

finrl_demo.py
from finrl.meta.preprocessor.yahoo_finance import YahooDownloader
from finrl.meta.env_stock_trading.env_stocktrading import StockTradingEnv

# 1. 下载苹果公司股票数据
df = YahooDownloader(
    start_date="2023-01-01",
    end_date="2023-12-31",
    ticker_list=["AAPL"]
).fetch_data()

# 2. 配置环境参数
stock_dimension = len(df.tic.unique())
env_kwargs = {
    "stock_dim": stock_dimension,
    "hmax": 100, 
    "initial_amount": 100000, 
    "reward_scaling": 1e-4
}

# 3. 创建环境
env = StockTradingEnv(df=df, **env_kwargs)

print("FinRL Environment Ready!")
查看 Github