bert文本情感分类、有数据代码、直接运行

 目录

导入库：

读取数据：

 数据清洗 ：

数据分析：词频统计 饼图

 数据分析：词频统计 柱状图

  数据分析：词频统计 词云

数据划分 训练集 和测试集 

 定义读取数据的函数

加载bert 定义模型

定义训练参数 epoch 优化器 学习率等

训练

模型加载预测

f1分数可以达到93

 所有的数据代码 请点击：

导入库：

import torch
from torch import nn
from torch import optim
import transformers as tfs
import math
import numpy as np
import pandas as pd
from sklearn.metrics import f1_score
import warnings
import re
import jieba
warnings.filterwarnings('ignore')
from tqdm import tqdm
from sklearn.model_selection import train_test_split
from collections import Counter
import matplotlib.pyplot as plt
plt.rcParams["font.sans-serif"] = ['Simhei']
plt.rcParams["axes.unicode_minus"] = False
from pylab import *

读取数据：

data = pd.read_excel('淮安评论合集.xlsx')
train_data=data["comment"]
train_label=data["sentiment"]
print(train_data.values)
print(train_label.values)

数据是带有标签的二分类的 01 标签