在TensorFlow 1.14.0 + Keras 2.2.5 on Python 3.6 中学习bilstm_crf模型时遇到的bug
看了很多博客都说把Embedding(len(vocab)+1, EMBED_DIM, mask_zero=True)中的“mask_zero=True”删除。但删除之后会影响到后续向量零扩展的效果。
观察报错信息
可知crf.py文件第516行出现类型问题,进行如下修改即可
mask2 = K.cast(K.concatenate([mask, K.zeros_like(mask[:, :1])], axis=1), | ||
mask2 = K.cast(K.concatenate([mask, K.cast(K.zeros_like(mask[:, :1]), mask.dtype)], axis=1), |