tf2crf
一个简单的CRF层用于tensorflow 2 keras
支持keras遮罩
安装
$ pip install tf2crf
特征
易于使用的带有张量流的CRF层
支持混合精度训练
支持具有DSC丢失的ModelWithCRFLossDSCLoss,这会在数据不平衡的情况下提高f1得分(请参阅)
注意力
在keras_contrib中添加CRF之类的内部内核,因此现在无需在CRF层之前堆叠Dense层。
我已经更改了将损耗函数和精度函数放在CRF层中的以前的方式。 相反,我选择使用ModelWappers(称为jaspersjsun),它更干净,更灵活。
尖端
tensorflow> = 2.1.0建议使用与您的tf版本兼容的最新tensorflow-addons。
例子
import tensorflow as tf
from tf2CRF import CRF
2023-04-09 18:10:30
8KB
Python
1