NER建立模型时出现concat tensors不匹配问题

本文详细解析了在建立NER模型时遇到的TypeError,深入分析了问题根源在于concatV2操作接受的不同类型张量,并给出了修改mask_zero参数的解决方案。

1 问题描述

    在建立NER识别模型时出现:

        TypeError: Tensors in list passed to 'values' of 'ConcatV2' Op have types [bool, float32] that don't all match.       

        

 

2 分析问题

    我们仔细看报错语句,可以看到真正报错的语句为:return tf.concat([to_dense(x) for x in tensors], axis)

        

    我们找到这行代码:

        

    进行断点调试:

          

    可以看到确实存在两个tensor的值类型不一致。

    回过头再来看报错的语句:

          TypeError: Tensors in list passed to 'values' of 'ConcatV2' Op have types [bool, float32] that don't all match.   

          这句话的意思是:传给concatV2操作的values值有两种类型,一直是bool类型,一种是float32类型。 

    从实际代码来看就是下面这一句:

        

    但是上面两个tensor值是哪里传过来的呢?

    经过调试,发现是从mask传过来的值:

        

    bool类型的值是mask,float32类型的值是zeros_like的返回值。

 

3 解决问题

    于是我们将建立模型时的参数:

        model.add(Embedding(len(self.vocab), self.Embedding_dim, mask_zero=True))

    改为:

        model.add(Embedding(len(self.vocab), self.Embedding_dim, mask_zero=False))

    于是,代码可以正常运行了:

     

 

4 问题总结

    (1) 需要认真看每一条报错信息对应代码;

    (2) 打断点调试到报错的语句,查看相应的数据值;

    (3) 怀疑是原始数据的问题导致,mask_zero不能设置为True,有待进一步分析。

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值