mlpack
Public Member Functions | List of all members
mlpack::ann::augmented::tasks::SortTask Class Reference

Generator of instances of the sequence sort task. More...

#include <sort.hpp>

Public Member Functions

 SortTask (const size_t maxLength, const size_t bitLen, bool addSeparator=false)
 Creates an instance of the sequence sort task. More...
 
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. More...
 
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. More...
 

Detailed Description

Generator of instances of the sequence sort task.

The parameters are:

Generated datasets are compliant with mlpack format - every dataset element is shaped as a vector of length (binary length) * (sequence length).

Example of generated dataset (sequence length = 3, binary length = 2):

Constructor & Destructor Documentation

◆ SortTask()

mlpack::ann::augmented::tasks::SortTask::SortTask ( const size_t  maxLength,
const size_t  bitLen,
bool  addSeparator = false 
)

Creates an instance of the sequence sort task.

Parameters
maxLengthMaximum length of the number sequence.
bitLenBinary length of sorted numbers.
addSeparatorFlag indicating whether generator should emit separating symbol after input sequence.

Member Function Documentation

◆ Generate() [1/2]

void mlpack::ann::augmented::tasks::SortTask::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.

Parameters
inputThe variable to store input sequences.
labelsThe variable to store output sequences.
batchSizeThe dataset size.
fixedLengthFlag indicating whether generator should emit sequences of pairwise equal length.

◆ Generate() [2/2]

void mlpack::ann::augmented::tasks::SortTask::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.

Parameters
inputThe variable to store input sequences.
labelsThe variable to store output sequences.
batchSizeThe dataset size.

The documentation for this class was generated from the following files: