doxygen/html/mlpack__utils_8hpp_source.html

#pragma once


#include <mlpack.hpp>


namespace netdem {


class MLPackUtils {

public:


  enum class LayerType {

    Identity,

    Linear,

    ReLU,

    LeakyReLU,

    ELU,

    Softmax,

    LogSoftMax

  };


  static double GetMSE(const arma::mat &pred, const arma::mat &Y);


  static double GetMAE(const arma::mat &pred, const arma::mat &Y);


  static arma::mat GetLabels(const arma::mat &ann_outputs);

};


} // namespace netdem

netdem::MLPackUtils
A class that defines several utility methods for ANN.
Definition mlpack_utils.hpp:10

netdem::MLPackUtils::LayerType
LayerType
An enumeration of neural network layer types.
Definition mlpack_utils.hpp:19

netdem::MLPackUtils::LayerType::Softmax
@ Softmax
The softmax layer type.

netdem::MLPackUtils::LayerType::Linear
@ Linear
The linear layer type.

netdem::MLPackUtils::LayerType::LogSoftMax
@ LogSoftMax
The log-softmax layer type.

netdem::MLPackUtils::LayerType::ELU
@ ELU
The exponential linear unit (ELU) layer type.

netdem::MLPackUtils::LayerType::LeakyReLU
@ LeakyReLU
The leaky ReLU layer type.

netdem::MLPackUtils::LayerType::Identity
@ Identity
The identity layer type.

netdem::MLPackUtils::LayerType::ReLU
@ ReLU
The rectified linear unit (ReLU) layer type.

netdem::MLPackUtils::GetMSE
static double GetMSE(const arma::mat &pred, const arma::mat &Y)
Calculates the mean squared error between predicted and ground truth data.
Definition mlpack_utils.cpp:6

netdem::MLPackUtils::GetLabels
static arma::mat GetLabels(const arma::mat &ann_outputs)
Converts continuous output vectors to class labels.
Definition mlpack_utils.cpp:28

netdem::MLPackUtils::GetMAE
static double GetMAE(const arma::mat &pred, const arma::mat &Y)
Calculates the mean absolute error between predicted and ground truth data.
Definition mlpack_utils.cpp:17

netdem
Definition bond_entry.hpp:7