RD-Agent Windows安装教程
QuantML QuantML 2024年09月23日 18:30
Content
RD-Agent 是微软亚洲研究院推出的一款自动化研究与开发工具,能够通过LLMs自动构建因子和策略,相关介绍见我们之前的文章:RD-Agent :自动化Quant工厂
然而,在安装RD-Agent时会遇到各种问题,尤其是目前官方还不支持win和macOS系统,本文介绍如何在win下安装RD-Agent。
首先官方github下载代码:
git clone https://github.com/microsoft/RD-Agent.git
创建新环境,安装所需的依赖:
conda create -n rdagent python=3.10
conda activate rdagent
pip install -e .
运行需要OPENAI API,可以使用官方API,没有的话也可以使用各种中转API,创建.env 文件:
cat << EOF > .env
OPENAI_API_KEY=<your_api_key>
# EMBEDDING_MODEL=text-embedding-3-small
CHAT_MODEL=gpt-4-turbo
EOF
RD-Agent也支持本地的llama模型,可以自行在hf下载权重文件。目前还不支持其他LLMs的API, 等后面有时间我们再增加对于其他大模型的支持。
接下来如果直接运行官方代码的话会遇到各种报错,主要是docker安装时的网络问题以及程序并行的问题。所以在运行之前,首先从星球获取修改后的Dockerfile,替换官方的rdagent\scenarios\qlib\docker\Dockerfile,接下来就可使用rd-agent自动搭建模型。
使用命令:
rdagent fin_model
LLMs即可自动思考,生成qlib格式的模型代码,自动运行,且不断改进模型。
输出结果:
这是运行两轮的结果,可以看到,通过LLMs构建的模型能够自动适配qlib并且输出结果,而且随着轮数增加,模型效果也会不断提升。也可以自己修改prompt 构造模型。