AI Hedge Fund 一个概念验证项目,目标是探索人工智能在交易决策中的应用。
该项目通过模拟多种投资策略的智能代理(agents),进行股票分析和交易决策。项目仅用于教育和研究目的,不适用于实际交易或投资。
主要功能
- Ben Graham Agent - 价值投资教父,只买有安全边际的隐藏宝石
- Bill Ackman 代理人 - 激进投资者,采取大胆的立场并推动变革
- Cathie Wood Agent - 成长型投资女王,相信创新和颠覆的力量
- 查理芒格经纪人 - 沃伦·巴菲特的合作伙伴,只以公道的价格购买精彩的企业
- Michael Burry 经纪人 - 寻找深度价值的大空头逆向者
- Peter Lynch 代理人 - 在日常业务中寻求“十倍”的实用投资者
- Phil Fisher Agent - 一丝不苟的成长型投资者,使用深入的“scuttlebutt”研究
- Stanley Druckenmiller 经纪人 - 寻找具有增长潜力的不对称机会的宏观传奇人物
- Warren Buffett 代理人 - 奥马哈的神谕,以合理的价格寻找优秀的公司
- 估值代理 - 计算股票的内在价值并生成交易信号
- Sentiment Agent - 分析市场情绪并生成交易信号
- Fundamentals Agent - 分析基本数据并生成交易信号
- Technicals Agent - 分析技术指标并生成交易信号
- Risk Manager - 计算风险指标并设置头寸限制
- 投资组合经理 - 做出最终交易决策并生成订单
免责声明
- 不用于真实交易或投资
- 不提供任何保证或担保
- 过去的表现并不代表未来的结果
- Creator 对经济损失不承担任何责任
- 咨询财务顾问进行投资决策
使用此软件,即表示您同意仅将其用于学习目的。
安装和使用
克隆存储库:
git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
- 安装 Poetry(如果尚未安装):
curl -sSL https://install.python-poetry.org | python3 -
- 安装依赖项:
poetry install
- 设置环境变量:
# Create .env file for your API keys
cp .env.example .env
- 设置 API 密钥:
# For running LLMs hosted by openai (gpt-4o, gpt-4o-mini, etc.)
# Get your OpenAI API key from https://platform.openai.com/
OPENAI_API_KEY=your-openai-api-key
# For running LLMs hosted by groq (deepseek, llama3, etc.)
# Get your Groq API key from https://groq.com/
GROQ_API_KEY=your-groq-api-key
# For getting financial data to power the hedge fund
# Get your Financial Datasets API key from https://financialdatasets.ai/
FINANCIAL_DATASETS_API_KEY=your-financial-datasets-api-key
重要提示:您必须设置 、 、 或 才能使对冲基金工作。如果要使用来自所有提供商的 LLM,则需要设置所有 API 密钥。OPENAI_API_KEYGROQ_API_KEYANTHROPIC_API_KEYDEEPSEEK_API_KEY
AAPL、GOOGL、MSFT、NVDA和TSLA的财务数据是免费的,不需要API密钥。
对于任何其他代码,您需要在 .env 文件中设置 。FINANCIAL_DATASETS_API_KEY
实战演习
场景1:经营对冲基金
poetry run python src/main.py --ticker AAPL,MSFT,NVDA
您还可以指定一个标志,以将每个代理的推理打印到控制台。--show-reasoning
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --show-reasoning
您可以选择指定开始日期和结束日期,以便为特定时间段做出决策。
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01
场景2:运行回溯测试器
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA
您可以选择指定要在特定时间段内进行回测的开始日期和结束日期。
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01