作者:老余捞鱼
原创不易,转载请标明出处及原作者。
写在前面的话:
本文介绍了如何利用 OpenAI 和 Python 进行股市情绪预测。主要通过使用 EODHD 提供的股市和金融新闻 API 来提取新闻数据,并利用 LangChain 和 OpenAI 的大型语言模型进行情感分析。
一、综述
在当今的股票市场上,随时了解新闻和事件对于做出战略决策至关重要。认识情绪对市场趋势的影响对于相应地调整战略至关重要。这一过程始于通过各种渠道获取大量市场新闻。其中最重要的是对数据质量(如来源数量、数据更新率等)和易用性的要求。
虽然数据可以在网上获得,而且很容易访问,但满足我们需求的最方便的方法之一是使用 API 端点,将市场数据和新闻直接集成到我们的代码中。提供 API 连接的金融数据提供商有很多,它们提供的数据包、支持方式和数据质量各不相同。
在本文中,我们将使用一家名为 EODHD 的市场数据提供商提供的股票市场和金融新闻 API。该 API 提供了一个从财经新闻中提取见解的端点,便于分析市场情绪。由于使用方便,用户可以查询和检索新闻文章,对市场的正面或负面情绪进行动态评估。
通过展示应用程序接口(API)的功能,我的目标是证明其与情绪分析的无缝整合,使我们能够根据当前的市场情绪做出明智的决策。在快节奏的股市环境中,有了这样一种资源,就能确保投资方法更具适应性和战略性。
二、导入软件包
首先,让我们将所需的软件包导入 Python 环境。我们将在本文中使用三个包,分别是用于处理数据帧的 pandas、用于提取数据的 eodhd 和用于构建 LLM 模型的 langchain。除此之外,我们还将使用 config 和 re 等其他辅助软件包:
!pip install openai
!pip install langchain
!pip install eodhd
!pip install configimport re
import requests
import pandas as pd
import config as cfg
from eodhd import APIClient
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain.chat_models import ChatOpenAI
在导入之前,确保使用命令行安装软件包。现在,我们已经将所有需要的软件包导入了 Python 环境,可以进行下一步,即激活 API 密钥。
三、激活 API 密钥
请首先访问其网站,然后完成注册过程以创建 EODHD 账户,最后导航到 "设置 "页面,在那里您可以找到您的 EODHD API 密钥。重要的是要确保这个秘密 API 密钥不会泄露给任何人。您可以按照以下代码激活 API 密钥:
api_key = '<YOUR API KEY>'
api = APIClient(api_key)
代码非常简单。在第一行,我们将 EODHD API 密钥存储到 api_key,然后在第二行,我们使用 eodhd 软件包提供的 APIClient 类激活 API 密钥,并将响应存储到客户端变量中。
请注意,您需要将 替换为您的 EODHD API 密钥。除了直接用文本存储 API 密钥外,还有其他方法可以提高安全性,例如利用环境变量等。
四、提取数据
我们将通过访问