BERT实战：实现多分类

最新推荐文章于 2024-08-31 21:21:20 发布

shun-ripking

最新推荐文章于 2024-08-31 21:21:20 发布

阅读量6.2k

点赞数 7

分类专栏：自然语言处理文章标签： bert 多分类

本文链接：https://blog.csdn.net/chengshunhang/article/details/110349236

版权

本文通过实战演示如何利用BERT进行文本多分类任务，具体案例是Kaggle上的真假新闻分类。首先加载数据，然后构建Dataset和DataLoader，接着在预训练的BERT模型基础上添加Linear分类器构建模型，最后进行训练和模型预测。

摘要由CSDN通过智能技术生成

前面以及介绍过bert的理论知识，以及它相应的实现方法，那么让我们通过实战加深对bert的了解。

我们将通过bert实现一个文本多分类任务，具体是kaggle上的一个真假新闻的任务。具体如下：

文件地址：https://www.kaggle.com/c/fake-news-pair-classification-
challenge/data 模型形式：BERT + Linear Classifier
参考链接：LeeMeng - 進擊的 BERT：NLP 界的巨人之力與遷移學習
参考博客：Simple to Bert | Ripshun Blog
github地址：nlp-code/Bert_真假新闻分类.ipynb at main · cshmzin/nlp-code (github.com)

加载数据

通过pandas将数据从csv中取出来

import pandas as pd
df_train = pd.read_csv("train.csv")
df_train = df_train.reset_index()
df_train = df_train.loc[:, ['title1_zh', 'title2_zh', 'label']]
df_train.columns = ['text_a', 'text_b', 'label']

df_test = pd.read_csv("test.csv")
df_test = df_test.loc[:, ["title1_zh", "title2_zh", "id"]]
df_test.columns = ["text_a", "text_b", "Id"]

print("训练样本数量：", len(df_train))
print("预测样本数：", len(df_test))
df_train.head()
df_test.head()