ChatGPT进行文本分类

1.建立客户端

python 复制代码
from openai import OpenAI
client = OpenAI(
        base_url="https://api.chatanywhere.tech/v1"
)

2.建立相关函数

python 复制代码
def get_openai_response(client, system_prompt, user_prompt, model="gpt-3.5-turbo"):
    response = client.chat.completions.create(
        model=model,
        messages=[
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": user_prompt}
        ],
    )
    return response.choices[0].message.content

3.设置分类文本

python 复制代码
q1 = "我刚买的XYZ智能手表无法同步我的日历,我应该怎么办?"
q2 = "XYZ手表的电池可以持续多久?"
q3 = "XYZ品牌的手表和ABC品牌的手表相比,有什么特别的功能吗?"
q4 = "安装XYZ智能手表的软件更新后,手表变得很慢,这是啥原因?"
q5 = "XYZ智能手表防水不?我可以用它来记录我的游泳数据吗?"
q6 = "我想知道XYZ手表的屏幕是什么材质,容不容易刮花?"
q7 = "请问XYZ手表标准版和豪华版的售价分别是多少?还有没有进行中的促销活动?"
q_list = [q1, q2, q3, q4, q5, q6, q7]

4.设置类别

python 复制代码
category_list = ["产品规格", "使用咨询", "功能比较", "用户反馈", "价格查询", "故障问题", "其它"]

5.设置提示模板

python 复制代码
classify_prompt_template = """
你的任务是为用户对产品的疑问进行分类。
请仔细阅读用户的问题内容,给出所属类别。类别应该是这些里面的其中一个:{categories}。
直接输出所属类别,不要有任何额外的描述或补充内容。
用户的问题内容会以三个#符号进行包围。

###
{question}
###
"""

6.分类

python 复制代码
for q in q_list:
    formatted_prompt = classify_prompt_template.format(categories=",".join(category_list), question=q)
    response = get_openai_response(client, formatted_prompt)
    print(response)

输出:

故障问题

产品规格

功能比较

故障问题

产品规格

产品规格

价格查询

相关推荐
JiKun几秒前
一键配置 Web 前端开发环境(PowerShell 自动化脚本)
前端·windows·程序员
yuuki2332335 分钟前
【C语言】预处理详解
c语言·windows·后端
墨尘笔尖2 小时前
使用子进程实现 C++ 与 Python 交互式控制台
c++·windows·python
许泽宇的技术分享2 小时前
Windows MCP.Net:解锁AI助手的Windows桌面自动化潜能
人工智能·windows·.net·mcp
IT小哥哥呀10 小时前
5 个 Windows 故障排除工具
windows·故障排除·系统运维·windows系统·电脑问题
应用市场11 小时前
构建自定义命令行工具 - 打造专属指令体
开发语言·windows·python
东方佑11 小时前
从字符串中提取重复子串的Python算法解析
windows·python·算法
Warren9816 小时前
复习MySQL
数据库·windows·tcp/ip·mysql·ubuntu·ssh·ansible
win4r17 小时前
🚀彻底改写浏览器自动化!ChatGPT Atlas浏览器深度评测:Agent模式自动操作网页太疯狂!跨网页理解+长文秒懂+图片识别,论文工作流被改写!看完你会
chatgpt·aigc·openai
低调小一20 小时前
KuiklyUI 科普:UI 如何映射到 Android View 并完成渲染
android·windows·ui