文件名称:tf2crf:张量流2 keras的CRF层
文件大小:8KB
文件格式:ZIP
更新时间:2024-06-04 19:25:25
Python
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
【文件预览】:
tf2crf-master
----tests()
--------test.py(3KB)
--------__init__.py(0B)
----tf2crf()
--------crf.py(3KB)
--------other_losses.py(382B)
--------model_wrapper.py(7KB)
--------__init__.py(112B)
--------requirements.txt(43B)
----LICENSE(1KB)
----setup.cfg(376B)
----setup.py(652B)
----Makefile(386B)
----README.md(2KB)