confusion_matrix_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_DATA_CONFUSION_MATRIX_IMPL_HPP
#define MLPACK_CORE_DATA_CONFUSION_MATRIX_IMPL_HPP

// In case it hasn't been included yet.
#include "confusion_matrix.hpp"

namespace mlpack {
namespace data {

template<typename eT>
void ConfusionMatrix(const arma::Row<size_t> predictors,
                     const arma::Row<size_t> responses,
                     arma::Mat<eT>& output,
                     const size_t numClasses)
{
  // Loop over the actual labels and predicted labels and add the count.
  output = arma::zeros<arma::Mat<eT> >(numClasses, numClasses);
  for (size_t i = 0; i < predictors.n_elem; ++i)
  {
    output.at(predictors[i], responses[i])++;
  }
}

} // namespace data
} // namespace mlpack

#endif