namespace mlpack {
namespace nn {

  class Autoencoder {
    public:
      Autoencoder(const arma::mat& data,
                  const size_t visibleSize,
                  const size_t hiddenSize,
                  NoiseLayerType&& noiseLayer = NoisetLayerType());

      // Set the size of the visible layer.
      void VisibleSize(const size_t visible) { this->visibleSize = visible; }

      // Get the size of the visible layer.
      size_t VisibleSize() const { return visibleSize; }

      // Sets size of the hidden layer.
      void HiddenSize(const size_t hidden) { this->hiddenSize = hidden; }

      // Gets the size of the hidden layer.
      size_t HiddenSize() const { return hiddenSize; }

      // Other functions like evaluate, gradient etc. here...

    private:
      const arma::mat& data;
      size_t visibleSize;
      size_t hiddenSize;
      NoiseLayerType = noiseLayer;
  };
} // nn
} // mlpack