mlpack 3.4.2
sort.hpp
Go to the documentation of this file.
1
14#ifndef MLPACK_METHODS_AUGMENTED_TASKS_SORT_HPP
15#define MLPACK_METHODS_AUGMENTED_TASKS_SORT_HPP
16
17#include <mlpack/prereqs.hpp>
18
19namespace mlpack {
20namespace ann /* Artificial Neural Network */ {
21namespace augmented /* Augmented neural network */ {
22namespace tasks /* Task utilities for augmented */ {
23
42{
43 public:
52 SortTask(const size_t maxLength,
53 const size_t bitLen,
54 bool addSeparator = false);
55
65 void Generate(arma::field<arma::mat>& input,
66 arma::field<arma::mat>& labels,
67 const size_t batchSize,
68 bool fixedLength = false) const;
77 void Generate(arma::mat& input,
78 arma::mat& labels,
79 const size_t batchSize) const;
80
81 private:
82 // Maximum length of the sequence.
83 size_t maxLength;
84 // Binary length of sorted numbers.
85 size_t bitLen;
86 // Flag indicating whether generator should produce
87 // separator as part of the sequence
88 bool addSeparator;
89};
90
91} // namespace tasks
92} // namespace augmented
93} // namespace ann
94} // namespace mlpack
95
96#include "sort_impl.hpp"
97#endif
Generator of instances of the sequence sort task.
Definition: sort.hpp:42
SortTask(const size_t maxLength, const size_t bitLen, bool addSeparator=false)
Creates an instance of the sequence sort task.
void Generate(arma::field< arma::mat > &input, arma::field< arma::mat > &labels, const size_t batchSize, bool fixedLength=false) const
Generate dataset of a given size.
void Generate(arma::mat &input, arma::mat &labels, const size_t batchSize) const
Generate dataset of a given size and store it in arma::mat object.
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: cv.hpp:1
The core includes that mlpack expects; standard C++ includes and Armadillo.