keras-contrib crf报错
Tensors in list passed to ‘values’ of ‘ConcatV2’ Op have types [bool, float32] that don’t all match.
方法一:(未尝试)
原因是因为keras版本问题,低于2.2.4版本不会报错,故可将版本降为2.2.4版本。
另外,我的tensorflow版本是1.13.1
方法二:(亲试可用)
也可以更改keras-contrib源码
根据报错信息找到crf文件,在代码前面加上
修改文件位于D:\Anaconda\envs\pyt1k2\Lib\site-packages\keras_contrib\layers\crf.py
mask2 = K.cast(K.concatenate([mask, K.zeros_like(mask[:, :1])], axis=1),
K.floatx())
改为
import tensorflow as tf
mask2 = K.cast(K.concatenate([tf.to_float(mask), K.zeros_like(mask[:, :1])], axis=1),
K.floatx())