如何使用大模型进行文本分类任务？_大模型分类任务

作者：喵喵爱编程 | 2024-07-31 18:46:18

踩

大模型分类任务

暑期实习基本结束了，校招即将开启。

不同以往的是，当前职场环境已不再是那个双向奔赴时代了。求职者在变多，HC 在变少，岗位要求还更高了。

最近，我们又陆续整理了很多大厂的面试题，帮助一些球友解惑答疑，分享技术面试中的那些弯弯绕绕。

总结链接如下：

《大模型实战宝典》（2024版）正式发布！

《大模型面试宝典》(2024版) 发布！

喜欢本文记得收藏、关注、点赞。更多实战和面试交流，文末加入我们

文本分类是自然语言处理（NLP）中的一项基础任务，应用范围从情感分析到内容分类。

传统上，它需要大量的数据预处理、特征工程和模型训练。大型语言模型的出现彻底改变了这一过程，提供了一种强大且高效的替代方案。

今天，我们将探讨如何使用大模型进行文本分类任务，并提供几个实际示例来展示如何实现它们。

大模型在文本分类中的优势

GPT-4o、Claude 3.5 等在文本分类方面有几个显著优势：

设置简便：大模型大大减少了对大量数据预处理和特征工程的需求。它们可以在没有特定领域特征训练的情况下理解文本的上下文和细微差别。
高性能：这些模型已经在海量数据上进行了预训练，使其能够在包括文本分类在内的许多NLP任务中达到最先进的性能。
少样本学习：通过在提示中注入少量示例，可以进一步提高性能。
多功能性：单个LLM可以通过微调或提示来执行各种文本分类任务，而不需要为每个任务准备单独的模型。
上下文理解：大模型擅长捕捉上下文信息，这对于复杂或模糊的文本分类尤其重要。

使用大模型实现文本分类

让我们看看如何使用大模型进行文本分类，并通过一些代码示例来说明。我们将通过OpenAI的GPT模型API展示二元分类和多类分类任务。

设置

首先，让我们设置我们的环境：

import openai
import os

# 设置你的OpenAI API密钥
openai.api_key = os.getenv("OPENAI_API_KEY")

def classify_text(prompt, max_tokens=100):
    response = openai.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": prompt}],
        max_tokens=max_tokens
    )
    return response.choices[0].message.content.strip()
1
2
3
4
5
6
7
8
9
10
11
12
13

(1) 二元分类

让我们从一个简单的情感分析任务开始，这是一个二元分类问题：

def sentiment_analysis(text):
    prompt = f"""
      将以下文本的情感分类为正面或负面：

      文本: "{text}"

      情感:
    """
    
    result = classify_text(prompt)
    return result

# 测试函数
sample_text = "我非常喜欢这部电影！演技出色，情节引人入胜。"
sentiment = sentiment_analysis(sample_text)
print(f"文本: {sample_text}")
print(f"情感: {sentiment}")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

这个示例展示了使用LLM进行二元分类的简便性。模型将根据输入文本的情感输出“正面”或“负面”。

(2) 多类分类

现在来看一个更具挑战性的多类分类案例，比如将新闻文章分类：

def categorize_news(text):
    prompt = f"""
      将以下新闻文章分类为以下类别之一：政治，科技，体育，娱乐，或商业。

      文章: "{text}"

      类别:
    """
    
    result = classify_text(prompt)
    return result

# 测试函数
news_article = "苹果今天宣布了其最新的iPhone型号，配备革命性的AI芯片和延长的电池寿命。"
category = categorize_news(news_article)
print(f"文章: {news_article}")
print(f"类别: {category}")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

这个示例展示了大模型如何轻松处理多类分类任务，从预定义列表中选择一个类别。

(3) 多标签分类

大模型还能够进行多标签分类，为单个文本应用多个标签：

def assign_tags(text):
    prompt = f"""
      为以下文本分配相关标签。从这个列表中选择尽可能多的标签：科技，AI，商业，创新，社交媒体，隐私，网络安全。

      文本: "{text}"

      标签:
    """
    
    result = classify_text(prompt, max_tokens=100)
    return result.split(", ")

# 测试函数
article = "Facebook的新AI算法旨在改进内容审核，同时关注用户隐私问题。"
tags = assign_tags(article)
print(f"文章: {article}")
print(f"标签: {tags}")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

这个示例展示了大模型如何为给定文本输出多个相关标签，使其适用于标记系统或多标签分类任务。

(4) 带置信度评分的分类

对于更细致的分类，我们可以要求LLM为每个类别提供置信度评分：

def classify_with_confidence(text):
    prompt = f"""
      将以下文本分类为以下类别之一：科技，商业，或政治。为每个类别提供置信度评分（0-100）。

      文本: "{text}"
      
      分类:
      科技: [评分]
      商业: [评分]
      政治: [评分]
    """
    result = classify_text(prompt, max_tokens=100)
    return result

# 测试函数
sample_text = "新贸易政策对国际运营的科技公司有重大影响。"
classification = classify_with_confidence(sample_text)
print(f"文本: {sample_text}")
print(f"分类:\n{classification}")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

这个示例展示了大模型如何提供更详细的分类结果，包括多个类别的置信度评分。

结论

大模型已经彻底改变了文本分类任务，为各种分类需求提供了灵活而强大的解决方案。正如这些示例所示，大模型可以通过最少的设置轻松处理二元、多类和多标签分类任务。

技术交流&资料

技术要学会分享、交流，不建议闭门造车。一个人可以走的很快、一堆人可以走的更远。

成立了算法面试和技术交流群，相关资料、技术交流&答疑，均可加我们的交流群获取，群友已超过2000人，添加时最好的备注方式为：来源+兴趣方向，方便找到志同道合的朋友。

方式①、微信搜索公众号：机器学习社区，后台回复：加群
方式②、添加微信号：mlc2040，备注：来自CSDN + 技术交流

用通俗易懂方式讲解系列

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/喵喵爱编程/article/detail/910275