virattt/ai-hedge-fund 是一个基于人工智能技术进行自动化投资决策的金融项目。简单讲，它是一个用Python构建的AI对冲基金团队，通过算法分析市场数据并执行交易。适用人群：量化交易爱好者、AI金融应用开发者、投资科技研究人员。

项目地址：https://github.com/virattt/ai-hedge-fund

主要语言：Python

stars: 40.77k

项目概述

这是一个由人工智能驱动的对冲基金概念验证项目，旨在探索如何利用人工智能进行交易决策。不过该项目仅用于教育和研究，不用于实际交易或投资。

核心功能

  • 多智能体协作：系统中有18个智能体协同工作，包括如Aswath Damodaran Agent（专注估值）、Ben Graham Agent（价值投资教父）、Warren Buffett Agent（寻找优质公司）等投资界知名人物对应的智能体，还有用于计算股票内在价值、分析市场情绪、基本面数据、技术指标的智能体，以及风险管理和投资组合管理智能体。
  • 交易信号生成：多个智能体从不同角度分析市场，如估值、情绪、基本面、技术指标等，生成交易信号。
  • 模拟决策：系统虽然不实际进行交易，但可以模拟交易决策过程。

优势

  • 教育价值高：对于想要学习人工智能在金融领域应用以及投资决策原理的人来说，是一个很好的学习工具。
  • 多视角分析：通过多个智能体从不同投资理念和分析角度进行决策，能提供更全面的市场分析。
  • 灵活运行方式：支持命令行界面和Web应用程序两种运行方式，满足不同用户的使用习惯。

应用场景

  • 金融学习：学生、投资者或金融爱好者可以通过该系统学习人工智能在投资决策中的应用，了解不同投资策略和分析方法。
  • 策略研究：研究人员可以利用该系统测试和验证新的投资策略和分析方法。

安装步骤

  • 克隆仓库：使用 git clone https://github.com/virattt/ai-hedge-fund.git 克隆项目，并进入项目目录。
  • 设置API密钥：创建并编辑 .env 文件，添加所需的API密钥，如 OPENAI_API_KEY 和 FINANCIAL_DATASETS_API_KEY。至少设置一个大语言模型（LLM）的API密钥，对于AAPL、GOOGL、MSFT、NVDA和TSLA之外的股票数据，需要设置金融数据集API密钥。

运行方式

  • 命令行界面：
    • 快速开始：安装Poetry并使用 poetry install 安装依赖。
    • 运行对冲基金：使用 poetry run python src/main.py --ticker AAPL,MSFT,NVDA 命令，还可以使用 -ollama 标志使用本地大语言模型，以及通过 -start-date 和 -end-date 指定决策的时间范围。
    • 运行回测器：使用 poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA 命令，同样支持 -ollama、-start-date 和 -end-date 标志。
  • Web应用程序：提供用户友好的界面，适合更喜欢可视化界面而非命令行工具的用户，详细安装和运行说明可参考此处。