极客前沿

AI 入门:一步步搭建你自己的文本分类工具

2026-06-06 01:32
DEV Tutorial
查看原文

零基础也能上手!本文教你安装环境、调用 AI 接口,完成第一个文本分类任务。

准备环境

你需要一台能上网的电脑,以及一个 Python(一种编程语言)环境。如果还没装,去 python.org 下载最新版,安装时勾选“Add Python to PATH”。打开终端(Mac/Linux)或命令提示符(Windows),输入 python --version,看到版本号就说明成功了。

安装必要的库

我们会用 OpenAI 的 API(一种让程序调用 AI 能力的方式)来分类文本。首先安装官方库:

  • 在终端运行 pip install openai,等待完成。
  • 然后安装 dotenv(用来管理密钥):pip install python-dotenv

常见坑:如果 pip 报错,试试 python -m pip install openai

获取 API 密钥

OpenAI 平台 注册账号,创建一个 API 密钥(一串密码)。复制密钥,在项目文件夹里新建一个 .env 文件,写入:OPENAI_API_KEY=你的密钥。注意别把这个文件上传到网上。

Tutorial Image

写第一段分类代码

新建一个 classify.py 文件,复制以下代码:

from openai import OpenAI
import os
from dotenv import load_dotenv

load_dotenv()
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))

response = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "system", "content": "将以下文本分类为:正面、负面或中性。"},
        {"role": "user", "content": "今天天气真好!"}
    ]
)

print(response.choices[0].message.content)

保存后在终端运行 python classify.py,如果看到“正面”之类的输出,就成功了!

验证与下一步

试着改 user 的内容为“好烦啊”,看看输出是不是“负面”。你也可以自定义分类规则,比如“垃圾邮件/正常邮件”。

下一步:可以把这个脚本做成一个 API 接口(让其他人通过网络调用),用 Flask 或 FastAPI 框架,部署到 RailwayRender 的免费套餐上。然后通过 Stripe 收费,就能卖你的分类服务了!

内容来源

DEV Tutorial

发布时间

2026-06-06 01:32

返回 AI技术