sort.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_AUGMENTED_TASKS_SORT_HPP
#define MLPACK_METHODS_AUGMENTED_TASKS_SORT_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace ann /* Artificial Neural Network */ {
namespace augmented /* Augmented neural network */ {
namespace tasks /* Task utilities for augmented */ {

class SortTask
{
 public:
  SortTask(const size_t maxLength,
           const size_t bitLen,
           bool addSeparator = false);

  void Generate(arma::field<arma::mat>& input,
                arma::field<arma::mat>& labels,
                      const size_t batchSize,
                      bool fixedLength = false) const;
  void Generate(arma::mat& input,
                arma::mat& labels,
                const size_t batchSize) const;

 private:
  // Maximum length of the sequence.
  size_t maxLength;
  // Binary length of sorted numbers.
  size_t bitLen;
  // Flag indicating whether generator should produce
  // separator as part of the sequence
  bool addSeparator;
};

} // namespace tasks
} // namespace augmented
} // namespace ann
} // namespace mlpack

#include "sort_impl.hpp"
#endif