用Keras做文本二分类,总是遇到如题错误,
我的类别是0或1,但是错误跟我说不能是1.
参见:Received a label value of 1 which is outside the valid range of [0, 1) - Python, Keras
loss function的问题。
原来用的是sparse_categorical_crossentropy,
改为binary_crossentropy问题解决。
本文探讨了在使用Keras进行文本二分类时,因loss函数选择不当导致的错误。原本采用的sparse_categorical_crossentropy不适用于二分类问题,更换为binary_crossentropy后问题得以解决。
用Keras做文本二分类,总是遇到如题错误,
我的类别是0或1,但是错误跟我说不能是1.
参见:Received a label value of 1 which is outside the valid range of [0, 1) - Python, Keras
loss function的问题。
原来用的是sparse_categorical_crossentropy,
改为binary_crossentropy问题解决。