- added StackCreator
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@823 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
@@ -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 */
|
||||
|
||||
Reference in New Issue
Block a user