
TF2CRF是TensorFlow 2.0与Keras CRF (条件随机场) 层的开源代码。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
tf2crf 提供了一个简洁的 CRF 层,专为 TensorFlow 2 和 Keras 设计,并支持 Keras 遮罩功能。通过 $ pip install tf2crf$ 进行安装。 这一层具备便捷的操作特性,并基于张量流构建,能够有效进行混合精度训练。此外,它还兼容具有 DSC (动态安全系数) 丢失的模型,从而在数据不平衡的场景下显著提升 F1 得分(详细信息请参考相关文档)。为了简化内部结构,在 Keras contrib 中添加了 CRF 类似功能的内核,因此现在无需在 CRF 层之前堆叠 Dense 层。 我已经调整了将损失函数和精度函数整合到 CRF 层中的先前方法。 相反地,我选择了使用 ModelWappers (由 jaspersjsun 开发),这种方案更具可读性、灵活性和清晰度。建议使用与您的 TensorFlow 版本兼容的最新 TensorFlow Addons 版本,特别是对于 TensorFlow >= 2.1.0 的版本。以下是一个示例:
import tensorflow as tffrom tf2CRF import CRF
全部评论 (0)
还没有任何评论哟~


