排课系统
排课系统
在线试用
排课系统
解决方案下载
排课系统
源码授权
排课系统
产品报价
25-8-04 07:51
import torch
from transformers import BertTokenizer, BertForSequenceClassification
# 加载预训练的BERT模型和分词器
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
model = BertForSequenceClassification.from_pretrained('bert-base-uncased')
# 示例输入
text = "我喜欢数学，特别是几何部分。"
# 对文本进行编码
inputs = tokenizer(text, return_tensors="pt")
# 模型预测
outputs = model(**inputs)
logits = outputs.logits
print(logits)
