使用大型语言模型进行情感分类
背景
这个提示测试大型语言模型的文本分类能力,通过提示它对一段文本进行分类。
提示
将文本分类为中性、消极或积极
文本:我觉得食物还可以。
情感:
提示模板
将文本分类为中 性、消极或积极
文本:{input}
情感:
代码 / API
- GPT-4 (OpenAI)
- Mixtral MoE 8x7B Instruct (Fireworks)
from openai import OpenAI
client = OpenAI()
response = client.chat.completions.create(
model="gpt-4",
messages=[
{
"role": "user",
"content": "将文本分类为中性、消极或积极\n文本:我觉得食物还可以。\n情感:\n"
}
],
temperature=1,
max_tokens=256,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
import fireworks.client
fireworks.client.api_key = "<FIREWORKS_API_KEY>"
completion = fireworks.client.ChatCompletion.create(
model="accounts/fireworks/models/mixtral-8x7b-instruct",
messages=[
{
"role": "user",
"content": "将文本分类为中性、消极或积极\n文本:我觉得食物还可以。\n情感:\n",
}
],
stop=["<|im_start|>","<|im_end|>","<|endoftext|>"],
stream=True,
n=1,
top_p=1,
top_k=40,
presence_penalty=0,
frequency_penalty=0,
prompt_truncate_len=1024,
context_length_exceeded_behavior="truncate",
temperature=0.9,
max_tokens=4000
)
Reference
- Prompt Engineering Guide (16 March 2023)