首页 > 解决方案 > 将 tfa.text.crf_sequence_score 扩展到波束搜索

问题描述

我有一个模型,它有一个 CRF 作为头层,并希望扩展https://www.tensorflow.org/addons/api_docs/python/tfa/text/crf_sequence_score以便像使用光束搜索一样获得多个假设。环顾四周,我找不到任何关于这方面的线索,阅读文档并不清楚。也许我可以把它放在一个 tf while 循环中,但即使这样,这个 API 也期望完整的潜力集来计算哪些在一步一步的 tf while 循环中不可用。我希望弄清楚如何做到这一点,或者是否可以使用这个 API。

标签: pythontensorflowneural-networkcrf

解决方案


推荐阅读