- integrated RnnStack
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@826 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+5
-1
@@ -54,6 +54,10 @@ public:
|
||||
StackType type();
|
||||
void setName(const std::string &name);
|
||||
size_t numLayers();
|
||||
virtual size_t numContext()
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
void addLayer(Layer *pLayer);
|
||||
void delLayer(Layer *pLayer);
|
||||
|
||||
@@ -67,7 +71,7 @@ public:
|
||||
|
||||
virtual void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) = 0;
|
||||
virtual void train(const arma::mat& batch, Rbm::IListener* pListener) = 0;
|
||||
arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch);
|
||||
virtual arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch) = 0;
|
||||
|
||||
size_t numTraining();
|
||||
void addTraining(const arma::mat &toAdd);
|
||||
|
||||
Reference in New Issue
Block a user