47 lines
1.2 KiB
C++
47 lines
1.2 KiB
C++
/*
|
|
* To change this license header, choose License Headers in Project Properties.
|
|
* To change this template file, choose Tools | Templates
|
|
* and open the template in the editor.
|
|
*/
|
|
|
|
/*
|
|
* File: DeepStack.hpp
|
|
* Author: jens
|
|
*
|
|
* Created on 25. Oktober 2019, 18:26
|
|
*/
|
|
|
|
#ifndef DEEPSTACK_HPP
|
|
#define DEEPSTACK_HPP
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <armadillo>
|
|
#include <jsoncpp/json/json.h>
|
|
#include "Layer.hpp"
|
|
#include "AStack.hpp"
|
|
|
|
class DeepStack : public AStack
|
|
{
|
|
public:
|
|
DeepStack(const std::string &name);
|
|
DeepStack(const DeepStack& orig) = delete;
|
|
virtual ~DeepStack();
|
|
|
|
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
|
|
arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch);
|
|
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener);
|
|
|
|
arma::mat upPass(size_t layerId, arma::mat const &v);
|
|
arma::mat downPass(size_t layerId, arma::mat const &h);
|
|
arma::mat upDownPass(size_t layerId, arma::mat const &v);
|
|
|
|
arma::mat upPass(Layer *pLayer, arma::mat const &v);
|
|
arma::mat downPass(Layer *pLayer, arma::mat const &h);
|
|
|
|
};
|
|
|
|
|
|
#endif /* DEEPSTACK_HPP */
|
|
|