为什么to_categorical为0等于[1,]而不等于[1,0]?

时间:2019-12-09 02:54:17

标签: keras

每当我在keras中尝试to_categorical函数时,即使我想要[1,0],它也总是输出[1,]。我该如何解决这个问题?

1 个答案:

答案 0 :(得分:2)

Keras无法知道您拥有的课程数量。如果仅传递一个0,则假定只有一个类。 要解决您的问题,请使用:

to_categorical(0,num_classes=2)