首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
  • 您找到你想要的搜索结果了吗?
    是的
    没有找到

    Keras中的多分类损失函数用法categorical_crossentropy

    = mnist.load_data() y_train = to_categorical(y_train, 10) y_test = to_categorical(y_test, 10) ... model.compile...分类交叉熵适用于多分类问题中,我的心电分类是一个多分类问题,但是我起初使用了二进制交叉熵,代码如下所示: sgd = SGD(lr=0.003, decay=0, momentum=0.7, nesterov=False) model.compile...所以就会出现我遇到的情况,这里引用了论坛一位大佬的样例: model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy...所以就造成了这个问题(其中的具体原理我也没去看源码详细了解) 解决 所以问题最后的解决方法就是: 对于多分类问题,要么采用 from keras.metrics import categorical_accuracy model.compile...(loss='binary_crossentropy', optimizer='adam', metrics=[categorical_accuracy]) 要么采用 model.compile(

    6.2K30
    领券