mse.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> mse_loss(const O& output, const L& labels, value_t<O> scale) {
    return scale * sum((output - labels) >> (output - labels));
}

template <typename O, typename L>
value_t<O> mse_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>> mse(const O& output, const L& labels, value_t<O> alpha, value_t<O> beta) {
    return std::make_pair(mse_loss(output, labels, alpha), mse_error(output, labels, beta));
}

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