用python写一个人工智能项目的代码
时间: 2023-09-24 10:11:25 浏览: 59
以下是一个简单的 Python 人工智能项目代码,基于 OpenAI 的 GPT-2 模型和 Transformers 库:
```python
import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel
# 加载 GPT-2 模型和 tokenizer
tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
model = GPT2LMHeadModel.from_pretrained('gpt2')
# 输入文本
text = "The quick brown fox jumps over the lazy dog."
# 转换为 token
input_ids = tokenizer.encode(text, return_tensors='pt')
# 生成文本
output = model.generate(input_ids, max_length=50, do_sample=True)
# 转换为字符串
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(output_text)
```
该代码使用 GPT-2 模型和 tokenizer 生成文本。首先,使用 tokenizer 将输入文本转换为 token,并将其传递给模型生成文本。在此示例中,我们将生成的文本长度限制为 50 个 token,并进行了采样。最后,使用 tokenizer 将生成的 token 转换为字符串并打印输出。