讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在Python中使用TensorFlow进行实体命名识别

胖浩同学_5637

胖浩同学_5637

发布时间:2026-11-04 07:11:24

|

625人浏览过

|

来源于php中文网

原创

TensorFlow不提供开箱即用的NER流水线,推荐使用Hugging Face的TFBertForTokenClassification;需注意subword与原始token的label对齐,首个subword分配标签、其余设为-100,并用offset_mapping处理边界。

如何在python中使用tensorflow进行实体命名识别

TensorFlow 本身不提供开箱即用的实体命名识别(NER)流水线,你需要自己构建模型或加载预训练模型;直接调用 tensorflow.keras.layers.BiLSTM 或 tf.keras.Model 搭建序列标注结构是可行路径,但更推荐用 Hugging Face 的 transformers 库加载 TFBertForTokenClassification ——它底层依赖 TensorFlow,且支持 fine-tuning 和推理。

为什么不用 tf.keras 自己从头写 CRF 层

CRF(条件随机场)层能提升 NER 标签序列一致性,但 TensorFlow 原生不带稳定可用的 CRF 实现;社区版 tfa.text.crf(来自 tensorflow-addons)存在梯度兼容问题,尤其在 TF 2.10+ 和 eager mode 下容易报 TypeError: Cannot convert a symbolic Tensor。除非你明确需要定制转移矩阵或复现某篇论文,否则绕过 CRF、用 softmax + 维特比后处理更稳妥。

实操建议:

  • 优先用 transformers 提供的 TFBertForTokenClassification,它已内置 token-level 分类头,输出 shape 是 (batch_size, seq_len, num_labels)
  • 若必须用纯 TF 实现,用 tf.keras.layers.Dense 接 BiLSTM 输出,loss 选 sparse_categorical_crossentropy,标签用整数编码(非 one-hot)
  • 避免在 model.compile() 中传入 tf.keras.losses.CategoricalCrossentropy(from_logits=True) —— 这会导致 logits 和 softmax 混用,引发 nan loss

如何正确对齐 subword 和原始 token 的 label

BERT 类模型会把词切分成 subword(如 "playing" → ["play", "##ing"]),但 NER 标签只标在原始 token 上,直接按 subword 分配 label 会导致错位。常见错误是把第一个 subword 标为实体起始(B-PER),其余标为 I-PER,却忽略后续 subword 对应原始 token 已结束。

立即学习“Python免费学习笔记(深入)”;

Python Pep Authoring
Python Pep Authoring

依据官方 python/peps仓库规则起草、审阅、编辑或准备 Python 改进提案(PEP),用于处理 PEP 文本和元数据头部。

下载

实操建议:

  • 用 tokenizer.convert_ids_to_tokens() 检查 subword 切分结果,确认 [CLS] 和 [SEP] 占位符位置
  • 对每个原始 token,用 tokenizer.encode_plus(..., return_offsets_mapping=True) 获取其在原文中的字符区间,再映射到 subword 索引
  • label 对齐时:仅给首个 subword 分配原始 token 的 label,其余 subword 统一设为 -100(PyTorch 风格 ignore_index,TF 中需在 loss 计算时 mask 掉)

加载 Hugging Face 模型并做 inference 的最小可行代码

以下代码片段可直接运行(需安装 transformers[tf-cpu] 和 tensorflow):

from transformers import TFBertForTokenClassification, AutoTokenizer
import tensorflow as tf
<p>model = TFBertForTokenClassification.from_pretrained("dslim/bert-base-NER")
tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")</p><p>text = "Apple Inc. is based in Cupertino."
inputs = tokenizer(text, return_tensors="tf", padding=True, truncation=True)</p><p>outputs = model(inputs)
predictions = tf.argmax(outputs.logits, axis=-1)[0].numpy()</p><h1>将预测 ID 映射回 label 名</h1><p>labels = model.config.id2label
tokens = tokenizer.convert_ids_to_tokens(inputs["input_ids"][0])
for token, pred_id in zip(tokens, predictions):
if token not in ["[CLS]", "[SEP]", "[PAD]"]:
print(f"{token} → {labels[pred_id]}")

注意:TFBertForTokenClassification 默认输出的是每个 subword 的 logits,不是原始句子 token 的;上面代码只是演示流程,实际部署需补全 offset 映射逻辑,否则 “Apple Inc.” 可能被拆成两个独立实体。

真正难的不是跑通 inference,而是 label 对齐和边界处理——尤其是中英文混合、缩写、标点粘连时,return_offsets_mapping 返回的 (start, end) 很容易漏掉空格或换行符,导致 token 切分偏移。这点在训练数据预处理阶段就该验证,别等到 eval 时发现 F1 掉点才回头查。

热门AI工具

更多
创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1851

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

5064

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1809

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

30517

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3627

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3647

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1243

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

Docker Desktop怎么设置端口映射
Docker Desktop怎么设置端口映射

本专题聚焦Docker Desktop容器端口映射与访问配置,整理从创建容器、设置端口绑定到验证本地服务访问的操作流程。针对端口被占用、映射配置错误、容器服务无法访问等情况,也提供相应的检查方向,方便用户完成端口配置。

0

2026.11.03

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn