cce.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> cce_loss(const O& output, const L& labels, value_t<O> scale) {
    return scale * etl::sum(log(output) >> labels);
}

template <typename O, typename L>
value_t<O> cce_error(const O& output, const L& labels, value_t<O> scale) {
    return scale * sum(min(abs(argmax(labels) - argmax(output)), 1.0));
}

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

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