Metrics.hpp

Go to the documentation of this file.

#pragma once
 
#include <Eigen/Dense>
#include "../Utilities/clue.hpp"
 
namespace NNFS
{
    class Metrics
    {
    public:
        static void accuracy(double &accuracy, const Eigen::MatrixXd &predicted,
                             const Eigen::MatrixXd &labels)
        {
            Eigen::VectorXi absolute_predictions;
            onehotdecode(absolute_predictions, predicted);
 
            Eigen::VectorXi class_labels;
            onehotdecode(class_labels, labels);
 
            accuracy = (absolute_predictions.array() == class_labels.array()).cast<double>().mean();
        }
 
        static void onehotdecode(Eigen::VectorXi &decoded, const Eigen::MatrixXd &onehot)
        {
            decoded.resize(onehot.rows());
            for (int i = 0; i < onehot.rows(); ++i)
            {
                onehot.row(i).maxCoeff(&decoded[i]);
            };
        }
    };
} // namespace NNFS