针对pytorch中的CRF不存在属性

为了更加便捷地使用CRF模块,有大佬真们对CRF封装起来,方便大家使用。关于CRF的左右,本博客不具体介绍,有兴趣的朋友可以百度查找。

本文查阅相关的博客,都没有人提到这个方法:

在使用CRF时,通常需要对CRF模块进行安装,CRF在python中具有多个封装形式

网上的第一种方式:不管用

常见的安装方法有

pip install TorchCRF

使用时方法

from TorchCRF import CRF

但是这种方法,存在一个问题,在CRF的应用中CRF的属性里面没有batch_first属性,也就是不能做到,在使用时将会报错。如下所示,该代码就会提示CRF没有batch_first属性。

class model(torch.nn.Model):
	def __init__(self, your_args_here, num_labels):
		# Your code here.
		self.crf = CRF(num_tags=config.num_labels, batch_first=True)

网上的第二种方式:不管用

pip install pytorch-crf

使用时方法

from torchcrf import CRF

但是在上面这条语句写出来时,会提示没有torchcrf这个库,无法往下面继续编写代码。
最后,想到了一个暂时性的解决方法

第三种方式:暂时性管用:

先去官网的下载地址,https://pypi.org/project/pytorch-crf/#files
下载箭头所指的文件,下载完成后得到一个压缩包。不要下载下面的whl文件,安装后还是无法解决问题。
针对pytorch中的CRF不存在属性_第1张图片

将文件解压后,打开文件目录:大致是这样子的
针对pytorch中的CRF不存在属性_第2张图片
打开第二个文件夹,torchcrf文件夹,里面会有一个__init__.py的文件
针对pytorch中的CRF不存在属性_第3张图片
将这个文件复制到你的项目文件,并将文件名改为你所需要的名字,例如torchcrf.py
在该文件的同级目录就可以使用以下命令使用CRF,这样的CRF就存在batch_first属性。

from torchcrf import CRF
crf = CRF(num_tags=config.num_labels, batch_first=True)

创作不易,如果对您有帮助的话,就帮忙点赞一下吧,不盛感激。
当然如果大家有更好的解决方法请在评论区解答一下!!!

你可能感兴趣的:(python,机器学习,pytorch,tensorflow)