score_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_AUGMENTED_TASKS_SCORE_IMPL_HPP
#define MLPACK_METHODS_AUGMENTED_TASKS_SCORE_IMPL_HPP

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

namespace mlpack {
namespace ann /* Artificial Neural Network */ {
namespace augmented /* Augmented neural network */ {
namespace scorers /* Scoring utilities for augmented */ {

template<typename MatType>
double SequencePrecision(arma::field<MatType> trueOutputs,
                         arma::field<MatType> predOutputs,
                         double tol)
{
  double score = 0;
  size_t testSize = trueOutputs.n_elem;
  if (trueOutputs.n_elem != predOutputs.n_elem)
  {
    std::ostringstream oss;
    oss << "SequencePrecision(): number of predicted sequences ("
        << predOutputs.n_elem << ") should be equal to the number "
        << "of ground-truth sequences ("
        << trueOutputs.n_elem << ")"
        << std::endl;
    throw std::invalid_argument(oss.str());
  }

  for (size_t i = 0; i < testSize; ++i)
  {
    arma::vec delta = arma::vectorise(arma::abs(
      trueOutputs.at(i) - predOutputs.at(i)));
    double maxDelta = arma::max(delta);
    if (maxDelta < tol)
    {
      score++;
    }
  }
  score /= testSize;
  return score;
}

} // namespace scorers
} // namespace augmented
} // namespace ann
} // namespace mlpack

#endif