bce.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

namespace etl::impl::standard {

template <typename O, typename L>
value_t<O> bce_loss(const O& output, const L& labels, value_t<O> scale) {
    return scale * sum((labels >> log(output)) + ((1.0 - labels) >> log(1.0 - output)));
}

template <typename O, typename L>
value_t<O> bce_error(const O& output, const L& labels, value_t<O> scale) {
    return scale * asum(labels - output);
}

template <typename O, typename L>
std::pair<value_t<O>, value_t<O>> bce(const O& output, const L& labels, value_t<O> alpha, value_t<O> beta) {
    return std::make_pair(bce_loss(output, labels, alpha), bce_error(output, labels, beta));
}

} //end of namespace etl::impl::standard