当前位置:   article > 正文

BERT 实战

BERT 实战

上一篇我们学习了 BERT 的原理,这里我们将它应用在 NER 实战上,我们在前面课程中也学习过 NER,这里我们用同样的数据集来看如何应用 BERT,大家可以与前面的模型进行比较。

另外推荐大家使用 Google 的云端 Jupyter Notebook 开发环境 Colab,自带 TensorFlow、Matplotlib、NumPy、Pandas 等深度学习基础库,还可以免费使用 GPU、TPU。

下面进入实战,本文所用的数据仍然是 Kaggle 上的一个 NER 数据集,可以将其中的ner_dataset.csv 下载到本地。

加载所需的库

首先我们需要安装 transformers,里面包括 NLP 中比较前沿的模型:BERT、GPT、XLNet 等等,支持 TensorFlow 和 PyTorch。

  1. !pip install transformers
  2. !pip install seqeval
  3. import pandas as pd
  4. import numpy as np
  5. from tqdm import tqdm, trange
  6. import torch
  7. from torch.utils.data i
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/857752?site
推荐阅读
相关标签
  

闽ICP备14008679号