下面是一个使用先前训练好的BERT模型进行情感预测的Python程序。程序将读取指定文件夹中的多个JSON文件,并对每个文件中的评论内容进行情感分析,最后输出每个课程的正面情感比例。
import os
import json
import torch
from transformers import BertTokenizer, BertForSequenceClassification
from datasets import Dataset
import numpy as np
# 设置设备
device = torch.d