jens
91cc83d9ba
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@586 b431acfa-c32f-4a4a-93f1-934dc6c82436
47 lines
979 B
C++
47 lines
979 B
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: Stack.hpp
|
|
* Author: jens
|
|
*
|
|
* Created on 25. Oktober 2019, 18:26
|
|
*/
|
|
|
|
#ifndef STACK_HPP
|
|
#define STACK_HPP
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <armadillo>
|
|
#include <jsoncpp/json/json.h>
|
|
#include "RbmLayer.hpp"
|
|
|
|
class Stack
|
|
{
|
|
public:
|
|
Stack(const std::string &prjname);
|
|
Stack(const Stack& orig);
|
|
virtual ~Stack();
|
|
|
|
void addLayer(RbmLayer *pLayer);
|
|
RbmLayer* getLayer(size_t layerId) const;
|
|
|
|
void train(size_t layerId, const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener);
|
|
void train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener);
|
|
void save(size_t numTraining);
|
|
void saveWeights();
|
|
|
|
private:
|
|
const std::string &m_prjname;
|
|
RbmLayer *m_pLayers;
|
|
|
|
};
|
|
|
|
|
|
#endif /* STACK_HPP */
|
|
|