- added StackCreator

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@823 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-17 18:40:50 +00:00
parent 64ef87879f
commit e55f8e2153
12 changed files with 284 additions and 104 deletions
+36
View File
@@ -0,0 +1,36 @@
/*
* 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: RnnStack.hpp
* Author: jens
*
* Created on 17. Januar 2022, 12:43
*/
#ifndef RNNSTACK_HPP
#define RNNSTACK_HPP
#include "AStack.hpp"
class RnnStack : public AStack
{
public:
RnnStack(const std::string &name);
RnnStack(const std::string &name, Json::Value &layers, LayerConstructor *pLayerConstructor=nullptr);
RnnStack(const RnnStack& orig) = delete;
virtual ~RnnStack();
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) override;
private:
};
#endif /* RNNSTACK_HPP */