首页 > 解决方案 > 如何将基于 scikit-kearn 的损失添加到神经网络?

问题描述

我想训练一个具有来自 AdaBoost 分类器的损失分量的神经网络。

myLoss(y_true, y_pred, adaClf, Z_):
    loss = BinaryCrossEntropy(y_true, y_pred) + MeanSquareError(adaClf.predict(Y_pred), Z_)
    return loss

adaClfscikit-learn 分类器在哪里。如何在 Keras 中实现这一点?

标签: tensorflowkeras

解决方案


抱歉,您不能将外部函数用作 tensorflow 中的损失。有了tf.map你可以使用外部函数,但没有梯度可以流过它,你的损失肯定需要梯度。所以你可能必须在 tensorflow 中实现分类器。也许对你有帮助。


推荐阅读