- trainingdata always contains context
- load / store training batch with context - on load: add context part to legacy training batches - removed Rbm::setBatch() git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@790 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+1
-7
@@ -126,12 +126,7 @@ public:
|
||||
m_bv.submat(0, 0, bv.n_rows-1, bv.n_cols-1) = bv;
|
||||
}
|
||||
|
||||
void setBatch(arma::mat const &batch)
|
||||
{
|
||||
m_batch = batch;
|
||||
}
|
||||
|
||||
void train(IListener *pListener=nullptr);
|
||||
void train(arma::mat const &batch, IListener *pListener=nullptr);
|
||||
|
||||
static arma::mat normalize(const arma::mat &hidden);
|
||||
const arma::mat& whv() const;
|
||||
@@ -196,7 +191,6 @@ protected:
|
||||
private:
|
||||
noise_gen_t m_noise;
|
||||
arma::mat m_whv;
|
||||
arma::mat m_batch;
|
||||
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user