如何用transformor进行英文转中文的机器翻译
时间: 2023-05-21 17:05:27 浏览: 107
基于transformer的机器翻译
使用 transformor 进行英文到中文的机器翻译,需要先安装好相应的库。接着,可以使用以下代码进行翻译:
```
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-zh")
tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-zh")
def translate(text):
inputs = tokenizer(text, return_tensors="pt", padding=True)
outputs = model.generate(**inputs)
decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
return decoded
```
输入要翻译的英文文本,即可得到对应的中文翻译结果。
阅读全文